diff --git a/requirements.txt b/requirements.txt
index 108b74c..807c9a3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,6 +7,7 @@ pytz>=2024.1  # Added for timezone support
 aiohttp>=3.9.3
 requests>=2.32.3
 playwright>=1.50.0
+html5lib>=1.1  # Added for HTML parsing
 
 # Search Engine
 duckduckgo-search>=7.4.3
diff --git a/tests/test_plan_exec_llm.py b/tests/test_plan_exec_llm.py
index 07f2a8c..bf54828 100644
--- a/tests/test_plan_exec_llm.py
+++ b/tests/test_plan_exec_llm.py
@@ -6,10 +6,8 @@
 from pathlib import Path
 import sys
 
-# Add the parent directory to the Python path so we can import the module
-sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from tools.plan_exec_llm import load_environment, read_plan_status, read_file_content, create_llm_client, query_llm
-from tools.plan_exec_llm import TokenUsage
+from tools.token_tracker import TokenUsage
 
 class TestPlanExecLLM(unittest.TestCase):
     def setUp(self):
diff --git a/tools/__init__.py b/tools/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tools/llm_api.py b/tools/llm_api.py
index 03c0178..fee4954 100644
--- a/tools/llm_api.py
+++ b/tools/llm_api.py
@@ -12,7 +12,7 @@
 from typing import Optional, Union, List
 import mimetypes
 import time
-from token_tracker import TokenUsage, APIResponse, get_token_tracker
+from .token_tracker import TokenUsage, APIResponse, get_token_tracker
 
 def load_environment():
     """Load environment variables from .env files in order of precedence"""
diff --git a/tools/plan_exec_llm.py b/tools/plan_exec_llm.py
index c0c2f61..9861924 100644
--- a/tools/plan_exec_llm.py
+++ b/tools/plan_exec_llm.py
@@ -7,7 +7,7 @@
 from dotenv import load_dotenv
 import sys
 import time
-from token_tracker import TokenUsage, APIResponse, get_token_tracker
+from .token_tracker import TokenUsage, APIResponse, get_token_tracker
 
 STATUS_FILE = '.cursorrules'