diff --git a/__init__.py b/__init__.py
index 00bb088..7e148ad 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,9 +1,9 @@
-from .textview_node import TextViewNode
-from .agent_node import AgentNode
-from .task_node import TaskNode
-from .crew_node import CrewNode
-from .list_task_node import TaskList
-from .list_agent_node import AgentList
+# from .textview_node import TextViewNode
+from .nodes.agent_node import AgentNode
+from .nodes.task_node import TaskNode
+from .nodes.crew_node import CrewNode
+from .nodes.list_task_node import TaskList
+from .nodes.list_agent_node import AgentList
 
 NODE_CLASS_MAPPINGS = {
     "📎CrewAI Agent": AgentNode,
diff --git a/agent_node.py b/nodes/agent_node.py
similarity index 100%
rename from agent_node.py
rename to nodes/agent_node.py
diff --git a/base_node.py b/nodes/base_node.py
similarity index 100%
rename from base_node.py
rename to nodes/base_node.py
diff --git a/crew_node.py b/nodes/crew_node.py
similarity index 100%
rename from crew_node.py
rename to nodes/crew_node.py
diff --git a/list_agent_node.py b/nodes/list_agent_node.py
similarity index 100%
rename from list_agent_node.py
rename to nodes/list_agent_node.py
diff --git a/list_task_node.py b/nodes/list_task_node.py
similarity index 100%
rename from list_task_node.py
rename to nodes/list_task_node.py
diff --git a/task_node.py b/nodes/task_node.py
similarity index 100%
rename from task_node.py
rename to nodes/task_node.py