diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py b/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py
new file mode 100644
index 0000000..2c15130
--- /dev/null
+++ b/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py
@@ -0,0 +1,12 @@
+from enum import Enum
+
+
+class DistFromObject(Enum):
+    """
+    Distance in fields from object.
+
+    """
+
+    LT_3 = 0,
+    GE_3_LT_8 = 1,
+    GE_8 = 2