diff --git a/README.md b/README.md
index ba3c474..900a89f 100644
--- a/README.md
+++ b/README.md
@@ -51,13 +51,13 @@ WORKSPACE_DIR="agent_workspace"
 
 ```python
 # Example usage
-from autohedge import AutomatedTradingSystem
+from autohedge import AutoFund
 
 # Define the stocks to analyze
 stocks = ["NVDA"]
 
 # Initialize the trading system with the specified stocks
-trading_system = AutomatedTradingSystem(stocks)
+trading_system = AutoFund(stocks)
 
 # Define the task for the trading cycle
 task = "Let's analyze nvidia to see if we should buy it, we have 50k$ in allocation"
@@ -124,7 +124,7 @@ class AutoHedgeOutput(BaseModel):
 AutoHedge can be configured through environment variables or initialization parameters:
 
 ```python
-trading_system = AutomatedTradingSystem(
+trading_system = AutoFund(
     name="CustomStrategy",
     description="My Trading Strategy",
     stocks=["NVDA", "AAPL"],
@@ -285,7 +285,7 @@ flowchart LR
 ### Class Structure
 ```mermaid
 classDiagram
-    class AutomatedTradingSystem {
+    class AutoFund {
         +String name
         +String description
         +List stocks
@@ -314,10 +314,10 @@ classDiagram
         +generate_order()
     }
     
-    AutomatedTradingSystem --> TradingDirector
-    AutomatedTradingSystem --> QuantAnalyst
-    AutomatedTradingSystem --> RiskManager
-    AutomatedTradingSystem --> ExecutionAgent
+    AutoFund --> TradingDirector
+    AutoFund --> QuantAnalyst
+    AutoFund --> RiskManager
+    AutoFund --> ExecutionAgent
 ```
 
 ## 🤝 Contributing
diff --git a/autohedge/__init__.py b/autohedge/__init__.py
index e69de29..ddf93ab 100644
--- a/autohedge/__init__.py
+++ b/autohedge/__init__.py
@@ -0,0 +1,3 @@
+from autohedge.main import AutoFund
+
+__all__ = ["AutoFund"]
\ No newline at end of file
diff --git a/autohedge/main.py b/autohedge/main.py
index 7158eb2..1d5287e 100644
--- a/autohedge/main.py
+++ b/autohedge/main.py
@@ -310,7 +310,7 @@ def analyze(self, stock: str, thesis: str) -> str:
             raise
 
 
-class AutomatedTradingSystem:
+class AutoFund:
     """
     Main trading system that coordinates all agents and manages the trading cycle.
 
diff --git a/example.py b/example.py
index 2431f9d..52eb80c 100644
--- a/example.py
+++ b/example.py
@@ -1,11 +1,11 @@
 # Example usage
-from autohedge.main import AutomatedTradingSystem
+from autohedge.main import AutoFund
 
 # Define the stocks to analyze
 stocks = ["NVDA"]
 
 # Initialize the trading system with the specified stocks
-trading_system = AutomatedTradingSystem(stocks)
+trading_system = AutoFund(stocks)
 
 # Define the task for the trading cycle
 task = "Let's analyze nvidia to see if we should buy it, we have 50k$ in allocation"
diff --git a/pyproject.toml b/pyproject.toml
index 27e4861..28606b2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "autohedge"
-version = "0.0.2"
+version = "0.0.3"
 description = "autohedge - TGSC"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]