Merge and remove unused imports.

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()

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__":