diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTFood.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTFood.py
deleted file mode 100644
index 3619f20..0000000
--- a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTFood.py
+++ /dev/null
@@ -1,9 +0,0 @@
-from src.AI.DecisionTrees.projectSpecificClasses.DTEntities.DTSurvivalEntity import DTSurvivalEntity
-from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
-from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
-
-
-class DTFood(DTSurvivalEntity):
-    def __init__(self, distanceFromPlayer: DistFromObject):
-        super().__init__(distanceFromPlayer)
-        self.classification = SurvivalClassification.FOOD
diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTRestPlace.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTRestPlace.py
deleted file mode 100644
index bb2543a..0000000
--- a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTRestPlace.py
+++ /dev/null
@@ -1,9 +0,0 @@
-from src.AI.DecisionTrees.projectSpecificClasses.DTEntities.DTSurvivalEntity import DTSurvivalEntity
-from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
-from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
-
-
-class DTRestPlace(DTSurvivalEntity):
-    def __init__(self, distanceFromPlayer: DistFromObject):
-        super().__init__(distanceFromPlayer)
-        self.classification = SurvivalClassification.REST
diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTSurvivalEntity.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTSurvivalEntity.py
index b8fbab8..a280547 100644
--- a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTSurvivalEntity.py
+++ b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTSurvivalEntity.py
@@ -1,8 +1,10 @@
 from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
+from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
 
 
 class DTSurvivalEntity:
     DistFromObjectFromPlayer: DistFromObject
 
-    def __init__(self, distanceFromPlayer: DistFromObject):
+    def __init__(self, distanceFromPlayer: DistFromObject, classification: SurvivalClassification):
+        self.classification = classification
         self.distanceFromPlayer = distanceFromPlayer
diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTWater.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTWater.py
deleted file mode 100644
index 4aaf1b3..0000000
--- a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTWater.py
+++ /dev/null
@@ -1,9 +0,0 @@
-from src.AI.DecisionTrees.projectSpecificClasses.DTEntities.DTSurvivalEntity import DTSurvivalEntity
-from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
-from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
-
-
-class DTWater(DTSurvivalEntity):
-    def __init__(self, distanceFromPlayer: DistFromObject):
-        super().__init__(distanceFromPlayer)
-        self.classification = SurvivalClassification.WATER