diff --git a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalAttributesDefinitions.py b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalAttributesDefinitions.py
index 169ec3b..96ccb79 100644
--- a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalAttributesDefinitions.py
+++ b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalAttributesDefinitions.py
@@ -4,6 +4,13 @@ from src.AI.DecisionTrees.projectSpecificClasses.PlayerStatsValue import PlayerS
 
 
 class SurvivalAttributesDefinitions:
+    """
+    This class contains all attributes definitions which will be used to create decision tree examples
+    and decision tree learning.
+
+    There are definitions of attributes like hunger, thirst, etc...
+    """
+
     attrsDefinitionsCount = 6
 
     hungerAttrDef = AttributeDefinition(0, "Hunger", [PlayerStatsValue.ZERO_TO_QUARTER,
diff --git a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalDTExample.py b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalDTExample.py
index 8d4264c..e7be748 100644
--- a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalDTExample.py
+++ b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalDTExample.py
@@ -7,6 +7,10 @@ from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import S
 
 
 class SurvivalDTExample(DecisionTreeExample):
+    """
+    This class will be used to create examples for decision trees with project specific attributes like hunger,
+    distance from food, etc.
+    """
     hungerVal: PlayerStatsValue
     thirstVal: PlayerStatsValue
     staminaVal: PlayerStatsValue