diff --git a/src/app/board/board.component.ts b/src/app/board/board.component.ts
index 17268ad..89146f9 100644
--- a/src/app/board/board.component.ts
+++ b/src/app/board/board.component.ts
@@ -37,10 +37,9 @@ export class BoardComponent implements OnInit {
         }
       });
     });
-    console.log(AllMoves);
-    const games = [];
-    games.push(AllMoves);
     const ai = new AiService();
+    const games = [];
+    games.push(ai.getMoves(AllMoves));
     ai.trainOnGames(games, (data)=>{ console.log(data); });
   }