diff --git a/README.md b/README.md
index e9a14fb..232f246 100644
--- a/README.md
+++ b/README.md
@@ -37,7 +37,7 @@ pip install async-web-search
 ### Example 1: Search across multiple sources
 
 ```python
-from web_search import WebSearch, WebSearchConfig, GoogleSearchConfig
+from web_search import WebSearch, WebSearchConfig
 
 config = WebSearchConfig(sources=["google", "arxiv"])
 results = await WebSearch(config).search("quantum computing")
@@ -48,7 +48,8 @@ print(results)
 ### Example 2: Google Search
 
 ```python
-from web_search import GoogleSearch, GoogleSearchConfig
+from web_search import GoogleSearchConfig
+from web_search.google import GoogleSearch
 
 config = GoogleSearchConfig(
     api_key="your_google_api_key",
@@ -64,7 +65,8 @@ for result in results:
 ### Example 3: Wikipedia Search
 
 ```python
-from web_search import WikipediaSearch, BaseConfig
+from web_search import BaseConfig
+from web_search.wikipedia import WikipediaSearch
 
 wiki_config = BaseConfig(max_results=5, max_preview_chars=500)
 results = await WikipediaSearch(wiki_config)._search("deep learning")
@@ -76,7 +78,8 @@ for result in results:
 ### Example 4: ArXiv Search
 
 ```python
-from web_search import ArxivSearch, BaseConfig
+from web_search import BaseConfig
+from web_search.arxiv import ArxivSearch
 
 arxiv_config = BaseConfig(max_results=3, max_preview_chars=800)
 results = await ArxivSearch(arxiv_config)._search("neural networks")
diff --git a/pyproject.toml b/pyproject.toml
index 26aa53e..30cef63 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "async-web-search"
 requires-python = ">=3.9"
-version = "0.2.0"
+version = "0.2.1"
 description = "Async web search library supporting Google, Wikipedia, and arXiv"
 
 readme = "README.md"
diff --git a/src/web_search/__init__.py b/src/web_search/__init__.py
index 8264a07..c30b0fc 100644
--- a/src/web_search/__init__.py
+++ b/src/web_search/__init__.py
@@ -8,11 +8,11 @@
 from .search import WebSearch
 
 __all__ = [
-    "WebSearch",
-    "WebSearchConfig",
-    "GoogleSearchConfig",
     "BaseConfig",
+    "BaseSearch",
+    "GoogleSearchConfig",
     "SearchSources",
     "SearchResult",
-    "BaseSearch",
+    "WebSearch",
+    "WebSearchConfig",
 ]