diff --git a/setup.py b/setup.py
index 83f164e..c59c3e1 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,14 @@
-from setuptools import setup, find_packages
+#!/usr/bin/env python
+"""Setup script for ScaffoldForge."""
+
+from setuptools import find_packages, setup
 
 setup(
     name="scaffoldforge",
     version="0.1.0",
-    packages=find_packages(),
+    py_modules=["scaffoldforge"],
+    packages=find_packages(exclude=["tests*"]),
+    python_requires=">=3.9",
     install_requires=[
         "click>=8.1.7",
         "pygithub>=2.3.0",
@@ -23,5 +28,4 @@ setup(
             "scaffoldforge=scaffoldforge.main:main",
         ],
     },
-    python_requires=">=3.9",
-)
+)
\ No newline at end of file