diff --git a/test.py b/test.py
index 1394e4d..07b9dfe 100644
--- a/test.py
+++ b/test.py
@@ -1,10 +1,8 @@
-from src.agents.agent_factory import *
-from src.agents.geneticAlgAgentJon import GeneticAlgAgentJM
-from src.game.TetrisGameManager import *
-from src.game.block import Block
-from src.game.tetris import *
+from src.agents.geneticAlgAgent import GeneticAgent, train_genetic_algorithm
+
+import numpy as np
+
+agents = train_genetic_algorithm(50)
 
-genAgent = GeneticAlgAgentJM()
 
-genAgent.number_of_selection(10)