diff --git a/main.py b/main.py
index b140767..e698e2e 100644
--- a/main.py
+++ b/main.py
@@ -13,6 +13,7 @@ if __name__ == "__main__":
     graphics = Graphics()
     waiter = Waiter(graphics)
     rabbit = Rabbit()
+    tree = DecisionTree()
 
     # init functions
     graphics.drawBackground(waiter.matrix)