diff --git a/main.py b/test_langgraph/__main__.py
similarity index 98%
rename from main.py
rename to test_langgraph/__main__.py
index 0a3945c..81b95ec 100644
--- a/main.py
+++ b/test_langgraph/__main__.py
@@ -12,7 +12,7 @@ import logging
 
 logger = logging.getLogger(__name__)
 
-import tools
+from . import tools
 
 #MODEL = "gemma3:27b"
 MODEL = "qwen3:latest"
diff --git a/tools.py b/test_langgraph/tools.py
similarity index 100%
rename from tools.py
rename to test_langgraph/tools.py