fixing blinking grid

main.py

@@ -5,6 +5,14 @@ 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
@@ -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__":