diff --git a/main.py b/main.py
index 08e822cbc0ffe687103d2ba02386e5a5c9714609..f41af679d5060c063a90245c4838f400cf9033e1 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,5 @@
-from src.tree import gen_tree
+from skill_tree.tree import TreeWalker
 
 if __name__ == '__main__':
-    gen_tree('data')
+    walker = TreeWalker("data", "web", "Web")
+    walker.walk()
\ No newline at end of file