diff --git a/stagehand/__init__.py b/stagehand/__init__.py
index 3a40ef9..de085b5 100644
--- a/stagehand/__init__.py
+++ b/stagehand/__init__.py
@@ -1,7 +1,7 @@
 from .agent import Agent
-from .client import Stagehand
 from .config import StagehandConfig, default_config
 from .handlers.observe_handler import ObserveHandler
+from .main import Stagehand
 from .metrics import StagehandFunctionName, StagehandMetrics
 from .page import StagehandPage
 from .schemas import (
diff --git a/stagehand/client.py b/stagehand/main.py
similarity index 98%
rename from stagehand/client.py
rename to stagehand/main.py
index 9077d15..386e1c7 100644
--- a/stagehand/client.py
+++ b/stagehand/main.py
@@ -39,17 +39,10 @@
 
 class Stagehand:
     """
-    Python client for interacting with a running Stagehand server and Browserbase remote headless browser.
-
-    Now supports automatically creating a new session if no session_id is provided.
-    You can provide a configuration via the 'config' parameter, or use individual parameters to override
-    the default configuration values.
+    Main Stagehand class.
     """
 
-    # Dictionary to store one lock per session_id
     _session_locks = {}
-
-    # Flag to track if cleanup has been called
     _cleanup_called = False
 
     def __init__(