diff --git a/src/app/ai.service.ts b/src/app/ai.service.ts
index aa2d45e..51c6ee7 100644
--- a/src/app/ai.service.ts
+++ b/src/app/ai.service.ts
@@ -18,9 +18,7 @@ flipX(arr) {
 };
 
 flipY(arr) {
-
-
- this.flipX(arr.slice().reverse());
+ return this.flipX(arr.slice().reverse());
 }
 // Creates a 1 hot of the diff
 showMove(first, second) {
@@ -113,6 +111,7 @@ async trainOnGames(games, setState) {
   });
 
   // Tensorfy!
+  console.log(AllX);
   const stackedX = tf.stack(AllX);
   const stackedY = tf.stack(AllY);
   await this.trainModel(model, stackedX, stackedY);