diff --git a/src/ai/fitness/__init__.py b/src/ai/fitness/__init__.py
new file mode 100644
index 0000000..2e9e82a
--- /dev/null
+++ b/src/ai/fitness/__init__.py
@@ -0,0 +1,3 @@
+from .fitness import calculate_fitness
+
+__all__ = ["calculate_fitness"]
diff --git a/src/ai/fitness.py b/src/ai/fitness/fitness.py
similarity index 94%
rename from src/ai/fitness.py
rename to src/ai/fitness/fitness.py
index b2ce090..dd8b6ef 100644
--- a/src/ai/fitness.py
+++ b/src/ai/fitness/fitness.py
@@ -5,8 +5,6 @@ import numpy as np
 from game import Game
 from utils import CONFIG
 
-from .log import log
-
 
 def calculate_fitness(game: Game) -> float:
     field = np.where(game.field != None, 1, 0)
@@ -15,10 +13,6 @@ def calculate_fitness(game: Game) -> float:
     return fitness
 
 
-def get_peaks(field: np.ndarray) -> int:
-    pass
-
-
 def get_row_transitions(field: np.ndarray) -> int:
     pass