diff --git a/Interface/__init__.py b/Interface/__init__.py
new file mode 100644
index 0000000..e07b40a
--- /dev/null
+++ b/Interface/__init__.py
@@ -0,0 +1 @@
+from .interface import *
diff --git a/Interface/helpers.py b/Interface/helpers.py
index 5c5b0f2..43e1ced 100644
--- a/Interface/helpers.py
+++ b/Interface/helpers.py
@@ -3,14 +3,18 @@ import pygame
 BLACK = 0, 0, 0
 WHITE = 255, 255, 255
 
+
 def rect(screen, color, position_x, position_y, size_x, size_y):
-  pygame.draw.rect(screen, color, (position_x,position_y,size_x,size_y))
+    pygame.draw.rect(screen, color, (position_x, position_y, size_x, size_y))
+
 
 def line(screen, color, x_1, y_1, x_2, y_2):
-  pygame.draw.line(screen, color, (x_1, y_1), (x_2, y_2))
+    pygame.draw.line(screen, color, (x_1, y_1), (x_2, y_2))
+
 
 def create_screen(size):
-  return pygame.display.set_mode(size)
+    return pygame.display.set_mode(size)
+
 
 def draw_board(screen, color, width, height, tiles_x, tiles_y):
     tiles_width = width / tiles_x
@@ -25,4 +29,3 @@ def draw_board(screen, color, width, height, tiles_x, tiles_y):
     for i in range(tiles_y-1):
         line(screen, color, 0, temp_y, width, temp_y)
         temp_y += tiles_height
-
diff --git a/Interface/interface.py b/Interface/interface.py
new file mode 100644
index 0000000..fe7971a
--- /dev/null
+++ b/Interface/interface.py
@@ -0,0 +1,21 @@
+import sys
+import pygame
+from .helpers import *
+
+
+def initialize_interface(tiles_x, tiles_y):
+    size = width, height = 800, 800
+    screen = create_screen(size)
+    pygame.display.set_caption('Epic AI Vacuum Cleaner')
+
+    while True:
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                sys.exit()
+
+        screen.fill(BLACK)
+        # rect(screen, WHITE, 50,50,90,90)
+
+        draw_board(screen, WHITE, width, height, tiles_x, tiles_y)
+
+        pygame.display.flip()
diff --git a/Interface/main.py b/Interface/main.py
deleted file mode 100644
index 7c030ea..0000000
--- a/Interface/main.py
+++ /dev/null
@@ -1,23 +0,0 @@
-import sys, pygame
-from helpers import *
-pygame.init()
-
-size = width, height = 1000, 1000
-screen = create_screen(size)
-
-tiles_x, tiles_y = 10, 10
-
-
-while True:
-    for event in pygame.event.get():
-        if event.type == pygame.QUIT: sys.exit()
-
-
-    screen.fill(BLACK)
-    # rect(screen, WHITE, 50,50,90,90)
-    
-    draw_board(screen, WHITE, width, height, tiles_x, tiles_y)
-
-
-
-    pygame.display.flip()
\ No newline at end of file
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..7ad26ad
--- /dev/null
+++ b/main.py
@@ -0,0 +1,5 @@
+import pygame
+from Interface import initialize_interface
+
+pygame.init()
+initialize_interface(10, 10)