diff --git a/sudoku_game/game.py b/sudoku_game/game.py
index d4789ed..ecc827c 100644
--- a/sudoku_game/game.py
+++ b/sudoku_game/game.py
@@ -1,5 +1,6 @@
 # Author: 4sushi
 
+from __future__ import annotations
 from typing import List, Tuple
 import random
 import copy
@@ -119,4 +120,3 @@ def is_game_won(self) -> bool:
     
     def get_difficulty(self) -> str:
         return self.DIFFICULTIES[self.difficulty]
-
diff --git a/sudoku_game/game_ui.py b/sudoku_game/game_ui.py
index 458d90d..876e52c 100644
--- a/sudoku_game/game_ui.py
+++ b/sudoku_game/game_ui.py
@@ -1,12 +1,13 @@
 # Author: 4sushi
-
+from __future__ import annotations
 import curses
 from sudoku_game.game import GameSudoku
-from typing import List, Optional, Dict
+from typing import List, Optional
 import sys
 from datetime import datetime, timedelta
 
 
+
 class GameUI:
 
     def __init__(self):