diff --git a/src/app/ai.service.ts b/src/app/ai.service.ts
index 51335d7..cad0c70 100644
--- a/src/app/ai.service.ts
+++ b/src/app/ai.service.ts
@@ -115,7 +115,7 @@ constructModel() {
 };
 
 async trainOnGames(games, setState) {
-  const model = this.getModel();
+  const model = this.constructModel();
   // model.dispose();
   let AllX = [];
   let AllY = [];
diff --git a/src/app/board/board.component.ts b/src/app/board/board.component.ts
index 082554d..566f130 100644
--- a/src/app/board/board.component.ts
+++ b/src/app/board/board.component.ts
@@ -18,6 +18,8 @@ export class BoardComponent implements OnInit {
   history: any[];
   ai = new AiService();
 
+  games = [];
+
   constructor() {}
 
   ngOnInit(): void {
@@ -62,9 +64,9 @@ export class BoardComponent implements OnInit {
       });
     });
     // const ai = new AiService();
-    const games = [];
-    games.push(this.ai.getMoves(AllMoves));
-    this.ai.trainOnGames(games, (data)=>{ console.log(data); });
+    
+    this.games.push(this.ai.getMoves(AllMoves));
+    this.ai.trainOnGames(this.games, (data)=>{ console.log(data); });
   }
 
   resetGame() {