diff --git a/dungeonbattle/bootstrap.py b/dungeonbattle/bootstrap.py
index 2c241c7..4c2c851 100644
--- a/dungeonbattle/bootstrap.py
+++ b/dungeonbattle/bootstrap.py
@@ -2,6 +2,7 @@ from dungeonbattle.game import Game
 from dungeonbattle.display.display_manager import DisplayManager
 from dungeonbattle.term_manager import TermManager
 
+
 class Bootstrap:
 
     @staticmethod
@@ -11,4 +12,4 @@ class Bootstrap:
             game.new_game()
             display = DisplayManager(term_manager.screen, game)
             game.display_refresh = display.refresh
-            game.run(term_manager.screen)
\ No newline at end of file
+            game.run(term_manager.screen)
diff --git a/dungeonbattle/display/display.py b/dungeonbattle/display/display.py
index 5633578..2e15a26 100644
--- a/dungeonbattle/display/display.py
+++ b/dungeonbattle/display/display.py
@@ -1,11 +1,15 @@
 import curses
 from typing import Any, Union
 
-from typing import Any
 from dungeonbattle.tests.screen import FakePad
 
 
 class Display:
+    x: int
+    y: int
+    width: int
+    height: int
+
     def __init__(self, screen: Any, pack: Any):
         self.screen = screen
         self.pack = pack
@@ -13,19 +17,19 @@ class Display:
     def newpad(self, height: int, width: int) -> Union[FakePad, Any]:
         return curses.newpad(height, width) if self.screen else FakePad()
 
-    def resize(self, y, x, height, width):
+    def resize(self, y: int, x: int, height: int, width: int) -> None:
         self.x = x
         self.y = y
         self.width = width
         self.height = height
 
-    def refresh(self, *args):
+    def refresh(self, *args) -> None:
         if len(args) == 4:
             self.resize(*args)
         self.display()
-    
-    def display(self):
-        pass
+
+    def display(self) -> None:
+        raise NotImplementedError
 
     @property
     def rows(self) -> int:
diff --git a/dungeonbattle/display/display_manager.py b/dungeonbattle/display/display_manager.py
index 37a2876..c4d3768 100644
--- a/dungeonbattle/display/display_manager.py
+++ b/dungeonbattle/display/display_manager.py
@@ -8,18 +8,20 @@ from dungeonbattle.game import Game, GameMode
 
 
 class DisplayManager:
-    
+
     def __init__(self, screen: Any, g: Game):
         self.game = g
         self.screen = screen
         pack = TexturePack.get_pack(self.game.settings.TEXTURE_PACK)
         self.mapdisplay = MapDisplay(screen, pack)
         self.statsdisplay = StatsDisplay(screen, pack)
-        self.mainmenudisplay = MainMenuDisplay(self.game.main_menu, screen, pack)
-        self.displays = [self.statsdisplay, self.mapdisplay, self.mainmenudisplay]
+        self.mainmenudisplay = MainMenuDisplay(self.game.main_menu,
+                                               screen, pack)
+        self.displays = [self.statsdisplay, self.mapdisplay,
+                         self.mainmenudisplay]
         self.update_game_components()
 
-    def update_game_components(self):
+    def update_game_components(self) -> None:
         for d in self.displays:
             d.pack = TexturePack.get_pack(self.game.settings.TEXTURE_PACK)
         self.mapdisplay.update_map(self.game.map)
@@ -28,11 +30,10 @@ class DisplayManager:
     def refresh(self) -> None:
         if self.game.state == GameMode.PLAY:
             self.mapdisplay.refresh(0, 0, self.rows * 4 // 5, self.cols)
-            self.statsdisplay.refresh(self.rows*4//5, 0, self.rows//5, self.cols)
+            self.statsdisplay.refresh(self.rows * 4 // 5, 0,
+                                      self.rows // 5, self.cols)
         if self.game.state == GameMode.MAINMENU:
-            self.mainmenudisplay.refresh(0,0,self.rows, self.cols)
-
-#        self.menudisplay.refresh(self.position)
+            self.mainmenudisplay.refresh(0, 0, self.rows, self.cols)
 
     def ensure_resized(self, *pads) -> bool:
         """
@@ -52,4 +53,4 @@ class DisplayManager:
 
     @property
     def cols(self) -> int:
-        return curses.COLS if self.screen else 42
\ No newline at end of file
+        return curses.COLS if self.screen else 42
diff --git a/dungeonbattle/display/mapdisplay.py b/dungeonbattle/display/mapdisplay.py
index 6438d36..0a07ec0 100644
--- a/dungeonbattle/display/mapdisplay.py
+++ b/dungeonbattle/display/mapdisplay.py
@@ -1,21 +1,18 @@
 #!/usr/bin/env python
-from typing import Any
-import curses
-
-from dungeonbattle.display.texturepack import TexturePack
 from dungeonbattle.entities.player import Player
 from dungeonbattle.interfaces import Map
 from .display import Display
 
+
 class MapDisplay(Display):
     player: Player
-    
+
     def __init__(self, *args):
         super().__init__(*args)
 
-    def update_map(self, m: Map):
+    def update_map(self, m: Map) -> None:
         self.map = m
-        self.pad = curses.newpad(m.height, m.width + 1)
+        self.pad = self.newpad(m.height, m.width + 1)
 
     def update_pad(self) -> None:
         self.pad.addstr(0, 0, self.map.draw_string(self.pack))
@@ -36,4 +33,4 @@ class MapDisplay(Display):
         pmincol = max(0, min(self.map.width, pmincol))
         self.pad.clear()
         self.update_pad()
-        self.pad.refresh(pminrow, pmincol, sminrow, smincol, smaxrow, smaxcol)
\ No newline at end of file
+        self.pad.refresh(pminrow, pmincol, sminrow, smincol, smaxrow, smaxcol)
diff --git a/dungeonbattle/display/menudisplay.py b/dungeonbattle/display/menudisplay.py
index 1c2cbc0..6bcf9fc 100644
--- a/dungeonbattle/display/menudisplay.py
+++ b/dungeonbattle/display/menudisplay.py
@@ -1,63 +1,63 @@
 from dungeonbattle.menus import Menu, MainMenu
-from typing import Any
 from .display import Display
-import curses
+
 
 class MenuDisplay(Display):
     position: int
-    
-    def __init__(self, *args) :
-        super().__init__(*args)
-        self.menubox = self.newpad(self.rows,self.cols)
 
-    def update_menu(self, menu: Menu):
+    def __init__(self, *args):
+        super().__init__(*args)
+        self.menubox = self.newpad(self.rows, self.cols)
+
+    def update_menu(self, menu: Menu) -> None:
         self.menu = menu
-        self.values = [ str(a) for a in menu.values ]
+        self.values = [str(a) for a in menu.values]
         self.trueheight = len(self.values)
         self.truewidth = max([len(a) for a in self.values])
 
-        #Menu values are printed in pad
-        self.pad = self.newpad(self.trueheight,self.truewidth+2)
-        for i in range(self.trueheight) :
-            self.pad.addstr(i,0,"  " + self.values[i])
-        
+        # Menu values are printed in pad
+        self.pad = self.newpad(self.trueheight, self.truewidth + 2)
+        for i in range(self.trueheight):
+            self.pad.addstr(i, 0, "  " + self.values[i])
 
     def update_pad(self) -> None:
         for i in range(self.trueheight):
-            self.pad.addstr(i,0," ")
-        self.pad.addstr(self.menu.position,0,">") #set a marker on the selected line
+            self.pad.addstr(i, 0, " ")
+        # set a marker on the selected line
+        self.pad.addstr(self.menu.position, 0, ">")
 
     def display(self) -> None:
-        if self.height-2>=self.menu.position-1 :
+        if self.height - 2 >= self.menu.position - 1:
             cornery = 0
-        elif self.height-2 >= self.trueheight-self.menu.position :
-            cornery = self.trueheight-self.height+2
-            
+        elif self.height - 2 >= self.trueheight - self.menu.position:
+            cornery = self.trueheight - self.height + 2
+
+        # Menu box
+        self.menubox.addstr(0, 0, "┏" + "━" * (self.width - 2) + "┓")
+        for i in range(1, self.height - 1):
+            self.menubox.addstr(i, 0, "┃" + " " * (self.width - 2) + "┃")
+        self.menubox.addstr(self.height - 1, 0,
+                            "┗" + "━" * (self.width - 2) + "┛")
 
-        #Menu box
-        self.menubox.addstr(0,0,"┏"+"━"*(self.width-2)+"┓")
-        for i in range(1,self.height-1) :
-            self.menubox.addstr(i,0,"┃"+" "*(self.width-2)+"┃")
-        self.menubox.addstr(self.height-1, 0, "┗"+"━"*(self.width-2)+"┛")
-        
         self.menubox.refresh(0, 0, self.y, self.x,
-                          self.height + self.y,
-                         self.width + self.x)
+                             self.height + self.y,
+                             self.width + self.x)
         self.update_pad()
-        self.pad.refresh(cornery, 0, self.y+1, self.x+2,
-                          self.height-1 + self.y,
-                         self.width-1 + self.x)
-    
+        self.pad.refresh(cornery, 0, self.y + 1, self.x + 2,
+                         self.height - 1 + self.y,
+                         self.width - 1 + self.x)
+
     @property
     def preferred_width(self) -> int:
         return self.truewidth + 6
-    
+
     @property
     def preferred_height(self) -> int:
         return self.trueheight + 2
 
+
 class MainMenuDisplay(Display):
-    def __init__(self, menu : MainMenu, *args) :
+    def __init__(self, menu: MainMenu, *args):
         super().__init__(*args)
         self.menu = menu
         self.pad = self.newpad(self.rows, self.cols)
@@ -67,13 +67,14 @@ class MainMenuDisplay(Display):
 
         self.menudisplay = MenuDisplay(self.screen, self.pack)
         self.menudisplay.update_menu(self.menu)
-        
+
     def display(self) -> None:
-        for i in range(len(self.title)) :
-            self.pad.addstr(4+i,self.width//2-len(self.title[0])//2-1,self.title[i])
-        self.pad.refresh(0,0,self.y,self.x,self.height,self.width)
+        for i in range(len(self.title)):
+            self.pad.addstr(4 + i, self.width // 2
+                            - len(self.title[0]) // 2 - 1, self.title[i])
+        self.pad.refresh(0, 0, self.y, self.x, self.height, self.width)
         menuwidth = min(self.menudisplay.preferred_width, self.width)
-        menuy, menux = len(self.title)+8, self.width//2-menuwidth//2-1
-        self.menudisplay.refresh(menuy, menux, min(self.menudisplay.preferred_height, self.height-menuy), menuwidth)
-        
-        
+        menuy, menux = len(self.title) + 8, self.width // 2 - menuwidth // 2 - 1
+        self.menudisplay.refresh(
+            menuy, menux, min(self.menudisplay.preferred_height,
+                              self.height - menuy), menuwidth)
diff --git a/dungeonbattle/display/statsdisplay.py b/dungeonbattle/display/statsdisplay.py
index 0b15414..6ec27bc 100644
--- a/dungeonbattle/display/statsdisplay.py
+++ b/dungeonbattle/display/statsdisplay.py
@@ -1,18 +1,16 @@
-from typing import Any
 from .display import Display
-import curses
 
 from dungeonbattle.entities.player import Player
 
 
 class StatsDisplay(Display):
     player: Player
-    
+
     def __init__(self, *args):
         super().__init__(*args)
         self.pad = self.newpad(self.rows, self.cols)
-    
-    def update_player(self, p: Player):
+
+    def update_player(self, p: Player) -> None:
         self.player = p
 
     def update_pad(self) -> None:
diff --git a/dungeonbattle/game.py b/dungeonbattle/game.py
index 9ed09ce..f0989c9 100644
--- a/dungeonbattle/game.py
+++ b/dungeonbattle/game.py
@@ -1,7 +1,6 @@
 import sys
 from typing import Any
 
-from .display.display import Display
 from .entities.player import Player
 from .interfaces import Map
 from .settings import Settings