diff --git a/UI/window.py b/UI/window.py
index 72a94b0..08a597e 100644
--- a/UI/window.py
+++ b/UI/window.py
@@ -1,8 +1,7 @@
 import pygame as pg
-import time
-from pygame.locals import *
 from UI.grid import Grid, Node
 
+
 class Window():
     def __init__(self, grid: Grid):
         pg.init()
diff --git a/main.py b/main.py
index ae9b0fa..1e319bf 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,3 @@
-import pygame as pg
-import numpy as np
-
 from UI.grid import Grid
 from UI.window import Window
 
@@ -9,7 +6,7 @@ def main():
     # initialize grid
     grid = Grid(20, 20)
     # initialize window
-    Window(grid) 
+    Window(grid)
 
 
 if __name__ == "__main__":