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 ed3952b121bd3c8f95afa9bac3974084db3d0b0e..78b4fc532e348dc390708475ec6c7d5242649774 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from src.tree import TreeWalker
+from skill_tree.tree import TreeWalker
 
 if __name__ == '__main__':
     walker = TreeWalker("data", "ai", "ai")
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