diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
new file mode 100644
index 0000000000000000000000000000000000000000..503b3fa0f71302263a5bc1bbb64a1af9e5d811a4
--- /dev/null
+++ b/.pre-commit-config.yaml
@@ -0,0 +1,6 @@
+repos:
+  - repo: git@codechina.csdn.net:csdn/skill_tree_hook.git
+    rev: 8cb76ab3b493582d3dab0021fb6fddfa28ec5a66
+    hooks:
+      - id: pre-commit
+        verbose: true
\ No newline at end of file
diff --git a/main.py b/main.py
index 08e822cbc0ffe687103d2ba02386e5a5c9714609..4de15a1e0cdee142624fd38ce06d3e41d6a9ac04 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()
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..930557801e86f8323d6f219034f2ea894985e211
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+skill-tree-parser~=0.0.5
\ No newline at end of file