diff --git a/Interface/vacuum_render.py b/Interface/vacuum_render.py
index 9a8e08f..29f1b52 100644
--- a/Interface/vacuum_render.py
+++ b/Interface/vacuum_render.py
@@ -1,12 +1,14 @@
-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:
-    grid.start_draw()
+    while True:
+        grid.start_draw()
 
-    grid.board(10, 10)
-    grid.circle(75, 75, 20, color=Colors.RED)
-    grid.circle(225, 175, 20, color=Colors.GREEN)
+        grid.board(10, 10)
+        grid.circle(75, 75, 20, color=Colors.RED)
+        grid.circle(225, 175, 20, color=Colors.GREEN)
 
-    grid.end_draw()
+        grid.end_draw()
diff --git a/main.py b/main.py
index 7ad26ad..45c51de 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,3 @@
-import pygame
-from Interface import initialize_interface
+from Interface.vacuum_render import initial_draw
 
-pygame.init()
-initialize_interface(10, 10)
+initial_draw()
\ No newline at end of file