diff --git a/swarms/tools/core/code_interpreter/session.py b/swarms/tools/core/code_interpreter/main.py
similarity index 99%
rename from swarms/tools/core/code_interpreter/session.py
rename to swarms/tools/core/code_interpreter/main.py
index aeefc4aa..4cf612cd 100644
--- a/swarms/tools/core/code_interpreter/session.py
+++ b/swarms/tools/core/code_interpreter/main.py
@@ -8,9 +8,6 @@ from dotenv import load_dotenv
 from io import BytesIO
 from typing import Any, List, Optional, Sequence, Tuple, Union, TYPE_CHECKING
 
-from codeboxapi import CodeBox
-from codeboxapi.schema import CodeBoxOutput
-from codeinterpreterapi.chains.remove_download_link import remove_download_link
 from pydantic import BaseModel, BaseSettings, root_validator
 
 from langchain.agents import AgentExecutor, BaseSingleActionAgent