Add __name__ == __main__ statement

week2/solution/code/vacuum_world.py

 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