diff --git a/airflow-wingman/__init__.py b/airflow-wingman/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/airflow-wingman/pyproject.toml b/airflow-wingman/pyproject.toml
index 7cbae6d..1455a0a 100644
--- a/airflow-wingman/pyproject.toml
+++ b/airflow-wingman/pyproject.toml
@@ -32,7 +32,7 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [tool.hatch.build.targets.wheel]
-packages = ["airflow-wingman"]
+packages = ["src/airflow_wingman"]
 
 [tool.ruff]
 line-length = 200
diff --git a/airflow-wingman/src/airflow_wingman/__init__.py b/airflow-wingman/src/airflow_wingman/__init__.py
new file mode 100644
index 0000000..a1c4dad
--- /dev/null
+++ b/airflow-wingman/src/airflow_wingman/__init__.py
@@ -0,0 +1,6 @@
+from importlib.metadata import version
+
+from airflow_wingman.plugin import WingmanPlugin
+
+__version__ = version("airflow-wingman")
+__all__ = ["WingmanPlugin"]
diff --git a/airflow-wingman/llms_models.py b/airflow-wingman/src/airflow_wingman/llms_models.py
similarity index 100%
rename from airflow-wingman/llms_models.py
rename to airflow-wingman/src/airflow_wingman/llms_models.py
diff --git a/airflow-wingman/plugin.py b/airflow-wingman/src/airflow_wingman/plugin.py
similarity index 100%
rename from airflow-wingman/plugin.py
rename to airflow-wingman/src/airflow_wingman/plugin.py
diff --git a/airflow-wingman/templates/wingman_chat.html b/airflow-wingman/src/airflow_wingman/templates/wingman_chat.html
similarity index 100%
rename from airflow-wingman/templates/wingman_chat.html
rename to airflow-wingman/src/airflow_wingman/templates/wingman_chat.html