diff --git a/neo_sapiens/__init__.py b/neo_sapiens/__init__.py
index e69de29..6f70c46 100644
--- a/neo_sapiens/__init__.py
+++ b/neo_sapiens/__init__.py
@@ -0,0 +1,25 @@
+from neo_sapiens.few_shot_prompts import (
+    data,
+    data1,
+    data2,
+    data3,
+    orchestrator_prompt_agent,
+)
+from neo_sapiens.hass_schema import (
+    AgentSchema,
+    ToolSchema,
+    parse_json_from_input,
+    create_agents,
+)
+
+__all__ = [
+    "data",
+    "data1",
+    "data2",
+    "data3",
+    "orchestrator_prompt_agent",
+    "AgentSchema",
+    "ToolSchema",
+    "parse_json_from_input",
+    "create_agents",
+]
diff --git a/few_shot_prompts.py b/neo_sapiens/few_shot_prompts.py
similarity index 100%
rename from few_shot_prompts.py
rename to neo_sapiens/few_shot_prompts.py
diff --git a/hass_schema.py b/neo_sapiens/hass_schema.py
similarity index 99%
rename from hass_schema.py
rename to neo_sapiens/hass_schema.py
index 801cb3d..6c5ed00 100644
--- a/hass_schema.py
+++ b/neo_sapiens/hass_schema.py
@@ -7,7 +7,7 @@
 from pydantic import BaseModel, Field
 from swarms import Agent, Anthropic, SwarmNetwork
 
-from few_shot_prompts import (
+from neo_sapiens.few_shot_prompts import (
     data,
     data1,
     data2,