diff --git a/mcp_codebase.py b/mcp_codebase.py
index c919d1e..a8d67f7 100644
--- a/mcp_codebase.py
+++ b/mcp_codebase.py
@@ -1,6 +1,8 @@
 import os
 import signal
 import json
+from collections import defaultdict, deque
+
 
 import re
 import subprocess