diff --git a/main.py b/main.py
index 100cf77..8508a75 100644
--- a/main.py
+++ b/main.py
@@ -4,6 +4,8 @@ import subprocess
 import readline
 import time
 import getpass
+import sys
+import re
 
 # 自动补全功能，基于系统命令
 def completer(text, state):