diff --git a/overcooked_simulator/main.py b/overcooked_simulator/main.py
index 168a240f80ef4f2284cec14dc0058e07708d0397..9200613745e73fc2d51294a4f76f1f6c50676a82 100644
--- a/overcooked_simulator/main.py
+++ b/overcooked_simulator/main.py
@@ -2,17 +2,22 @@ from overcooked_simulator.player import Player
 import sys
 from overcooked_simulator.simulation_runner import Simulator
 from overcooked_simulator.overcooked_environment import Environment
+from overcooked_simulator.pygame_gui import PyGameGUI
+
 
 def main():
     simulator = Simulator(Environment, 300)
-
     simulator.register_player(Player("p1", [100, 200]))
     simulator.register_player(Player("p2", [200, 100]))
 
-    simulator.start()
+    gui = PyGameGUI(simulator)
 
+    simulator.start()
     print(simulator.get_state())
 
+    gui.start_pygame()
+
+
     simulator.stop()
     sys.exit()