Interface/vacuum_render.py

@@ -1,8 +1,10 @@
-from grid_draw import GridDraw, Colors
+from Interface.grid_draw import GridDraw, Colors
 
-grid = GridDraw(500, 500)
+# dummy function
+def initial_draw():
+    grid = GridDraw(500, 500)
 
-while True:
+    while True:
         grid.start_draw()
 
         grid.board(10, 10)

main.py

@@ -1,5 +1,3 @@
-import pygame
+from Interface.vacuum_render import initial_draw
-from Interface import initialize_interface
 
-pygame.init()
+initial_draw()
-initialize_interface(10, 10)