diff --git a/week2/solution/code/vacuum_world.py b/week2/solution/code/vacuum_world.py
index 3d9656efb05248a784801735e40abbaa6ce43b4f..94e83d0e9371e7a3a050df60f22a02da1a140569 100644
--- a/week2/solution/code/vacuum_world.py
+++ b/week2/solution/code/vacuum_world.py
@@ -1,5 +1,6 @@
 from une_ai.vacuum import VacuumGame, DISPLAY_HEIGHT, DISPLAY_WIDTH
 from agent_programs import simple_reflex_behaviour, model_based_reflex_behaviour, goal_based_reflex_behaviour, utility_based_reflex_behaviour
 
-# To test the different agent programs, change the function passed as parameter to create the instance of VacuumGame
-game = VacuumGame(utility_based_reflex_behaviour, DISPLAY_WIDTH, DISPLAY_HEIGHT)
\ No newline at end of file
+if __name__ == "__main__":
+    # To test the different agent programs, change the function passed as parameter to create the instance of VacuumGame
+    game = VacuumGame(utility_based_reflex_behaviour, DISPLAY_WIDTH, DISPLAY_HEIGHT)
\ No newline at end of file