diff --git a/main.py b/main.py
index 73ebbd7..90191dc 100644
--- a/main.py
+++ b/main.py
@@ -67,8 +67,8 @@ def main():
             if event.type == pygame.QUIT:
                 run = False
         #keys_pressed = pygame.key.get_pressed()
-        
-        steps = bfs(State(None, None, 0, 0, 'E'), 450, 100)
+        draw_window(agent, fields)
+        steps = bfs(State(None, None, 0, 0, 'E'), 250, 100)
         for interm in steps:
             if interm.action == 'LEFT':
                 agent.direction = (agent.direction - 1) % 4