Poprawka w odświeżaniu mapy

main.py

@@ -2,6 +2,7 @@ import pygame
 import warehouse
 import agent
 import random
+import sys
 from attributes import PackSize, PackStatus
 
 WINDOW_SIZE = (600, 600)
@@ -31,10 +32,14 @@ class MainGameFrame:
         self.agent = agent.Agent(starting_x, starting_y, self.warehouse_map, 20)
 
     def run(self):
+        while True:
+            for event in pygame.event.get():
+                if event.type == pygame.QUIT:
+                    pygame.quit()
+                    sys.exit()
             self.draw_floor()
             self.draw_packages()
             self.draw_agent()
-        while True:
             pygame.display.update()
 
     def draw_floor(self):