diff --git a/main.py b/main.py
new file mode 100644
index 00000000..5f044237
--- /dev/null
+++ b/main.py
@@ -0,0 +1 @@
+from scripts.main import main
\ No newline at end of file
diff --git a/scripts/__init__.py b/scripts/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/scripts/browse.py b/scripts/browse.py
index b0c745ef..e991a73c 100644
--- a/scripts/browse.py
+++ b/scripts/browse.py
@@ -1,7 +1,7 @@
 import requests
 from bs4 import BeautifulSoup
-from config import Config
-from llm_utils import create_chat_completion
+from .config import Config
+from .llm_utils import create_chat_completion
 
 cfg = Config()
 
diff --git a/scripts/call_ai_function.py b/scripts/call_ai_function.py
index f8238658..0a2343bc 100644
--- a/scripts/call_ai_function.py
+++ b/scripts/call_ai_function.py
@@ -1,8 +1,8 @@
-from config import Config
+from .config import Config
 
 cfg = Config()
 
-from llm_utils import create_chat_completion
+from .llm_utils import create_chat_completion
 # This is a magic function that can do anything with no-code. See
 # https://github.com/Torantulino/AI-Functions for more info.
 def call_ai_function(function, args, description, model=None):
diff --git a/scripts/json_parser.py b/scripts/json_parser.py
index e96c43f6..a03c5f5e 100644
--- a/scripts/json_parser.py
+++ b/scripts/json_parser.py
@@ -1,8 +1,8 @@
 import json
 from typing import Any, Dict, Union
-from call_ai_function import call_ai_function
-from config import Config
-from json_utils import correct_json
+from .call_ai_function import call_ai_function
+from .config import Config
+from .json_utils import correct_json
 
 cfg = Config()
 
diff --git a/scripts/json_utils.py b/scripts/json_utils.py
index 80aab192..9ab776f5 100644
--- a/scripts/json_utils.py
+++ b/scripts/json_utils.py
@@ -1,6 +1,6 @@
 import re
 import json
-from config import Config
+from .config import Config
 
 cfg = Config()
 
diff --git a/scripts/llm_utils.py b/scripts/llm_utils.py
index 94ba5f13..a4c50944 100644
--- a/scripts/llm_utils.py
+++ b/scripts/llm_utils.py
@@ -1,5 +1,5 @@
 import openai
-from config import Config
+from .config import Config
 cfg = Config()
 
 openai.api_key = cfg.openai_api_key
diff --git a/tests.py b/tests.py
new file mode 100644
index 00000000..ce21c1f4
--- /dev/null
+++ b/tests.py
@@ -0,0 +1,8 @@
+import unittest
+
+if __name__ == "__main__":
+    # Load all tests from the 'scripts/tests' package
+    suite = unittest.defaultTestLoader.discover('scripts/tests')
+    
+    # Run the tests
+    unittest.TextTestRunner().run(suite)
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/tests/context.py b/tests/context.py
new file mode 100644
index 00000000..873ee0a5
--- /dev/null
+++ b/tests/context.py
@@ -0,0 +1,4 @@
+import sys
+import os
+sys.path.insert(0, os.path.abspath(
+    os.path.join(os.path.dirname(__file__), '..')))
\ No newline at end of file
diff --git a/tests/test_browse_scrape_text.py b/tests/test_browse_scrape_text.py
index bfc3b425..84e8fa27 100644
--- a/tests/test_browse_scrape_text.py
+++ b/tests/test_browse_scrape_text.py
@@ -2,7 +2,6 @@
 # Generated by CodiumAI
 
 import requests
-import pytest
 
 from scripts.browse import scrape_text
 
diff --git a/tests/json_tests.py b/tests/test_json_parser.py
similarity index 97%
rename from tests/json_tests.py
rename to tests/test_json_parser.py
index fdac9c2f..99d57772 100644
--- a/tests/json_tests.py
+++ b/tests/test_json_parser.py
@@ -1,9 +1,6 @@
 import unittest
-import os
-import sys
-# Probably a better way:
-sys.path.append(os.path.abspath('../scripts'))
-from json_parser import fix_and_parse_json
+
+from scripts.json_parser import fix_and_parse_json
 
 class TestParseJson(unittest.TestCase):
     def test_valid_json(self):