diff --git a/setup.py b/setup.py
index bb0bf16..a75ab5e 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,16 @@
 setup(
     name="neurorift",
     version="1.0.0",
-    packages=find_packages(include=['modules', 'modules.*', 'utils', 'utils.*', 'ai_wrapper', 'ai_wrapper.*']),
+    packages=find_packages(
+        include=[
+            "modules",
+            "modules.*",
+            "utils",
+            "utils.*",
+            "ai_wrapper",
+            "ai_wrapper.*",
+        ]
+    ),
     py_modules=[
         # Top-level modules used by the CLI
         "neurorift_main",
@@ -13,9 +22,9 @@
         "ai_controller",
     ],
     package_data={
-        'modules': ['**/*.json', '**/*.md'],
-        'prompts': ['**/*.md', '**/*.txt'],
-        'configs': ['*.json'],
+        "modules": ["**/*.json", "**/*.md"],
+        "prompts": ["**/*.md", "**/*.txt"],
+        "configs": ["*.json"],
     },
     include_package_data=True,
     install_requires=[
@@ -76,4 +85,4 @@
     ],
     python_requires=">=3.10",
     keywords="security penetration-testing bug-bounty ai multi-agent reconnaissance vulnerability-scanning",
-)
\ No newline at end of file
+)