fixing blinking grid

main.py

@@ -5,13 +5,21 @@ pygame.init()
 screen = pygame.display.set_mode((1280, 720))
 c = (0, 150, 0)
 
+
+def draw_grid():
+    for y in range(80, 720, 80):  # horizontal lines
+        pygame.draw.line(screen, c, (80, y), (1280 - 80, y), 1)
+    for x in range(80, 1280, 80):  # vertical lines
+        pygame.draw.line(screen, c, (x, 80), (x, 720 - 80), 1)
+
+
 class Wozek:
     def __init__(self):
         self.x = 55
         self.y = 55
         self.height = 64
         self.width = 64
-        self.image = pygame.image.load("wozek.png") 
+        self.image = pygame.image.load("wozek.png")
         # Credit: Forklift icons created by Smashicons - Flaticon
         # https://www.flaticon.com/free-icons/forklift
 
@@ -29,7 +37,7 @@ def main():
                 sys.exit(0)
             elif event.type == pygame.KEYDOWN and event.key == pygame.K_DOWN:
                 if wozek.y <= 600:
-                        wozek.y += 80
+                    wozek.y += 80
             elif event.type == pygame.KEYDOWN and event.key == pygame.K_UP:
                 if wozek.y >= 100:
                     wozek.y -= 80
@@ -40,17 +48,11 @@ def main():
                 if wozek.x >= 100:
                     wozek.x -= 80
 
-
-        for x in range(0, 1280, 80):
-            pygame.draw.line(screen, c, (1, x), (1280, x), 2)
-            pygame.draw.line(screen, c, (x, 1), (x, 720), 2)
-            pygame.display.update()
-    
-
-        #Drawing
-        screen.fill((0,0,0)) #removes object trail
-        pygame.display.flip()   #updating frames
+        # Drawing
+        screen.fill((0, 0, 0))  # removes object trail
+        draw_grid()
         wozek.draw()
+        pygame.display.flip()  # updating frames
 
 
 if __name__ == "__main__":