diff --git a/swarms/swarms/swarms.py b/swarms/swarms/swarms.py
index 558b825d..fdd1089d 100644
--- a/swarms/swarms/swarms.py
+++ b/swarms/swarms/swarms.py
@@ -2,10 +2,7 @@ import asyncio
 import logging
 from typing import Optional
 
-from langchain import LLMChain, OpenAI, PromptTemplate
-
-
-from swarms.agents.tools.main import WebpageQATool, process_csv
+from langchain import OpenAI
 from swarms.boss.boss_node import BossNode
 from swarms.workers.worker_node import WorkerNode
 
diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py
index bbc7e415..b8f9fa08 100644
--- a/swarms/workers/worker_node.py
+++ b/swarms/workers/worker_node.py
@@ -6,6 +6,7 @@ from langchain.agents import Tool
 from langchain.chat_models import ChatOpenAI
 from langchain.docstore import InMemoryDocstore
 from langchain.embeddings import OpenAIEmbeddings
+# TODO: refactor to Agent class
 from langchain.experimental.autonomous_agents.autogpt.agent import AutoGPT
 from langchain.vectorstores import FAISS