diff --git a/src/reddit.py b/src/reddit.py
index f800fdf..31cac34 100644
--- a/src/reddit.py
+++ b/src/reddit.py
@@ -197,19 +197,4 @@ class Tools:
 		await __event_emitter__({ "data": { "description": f"Starting retrieval for {page_type.value}/{id}...", "status": "in_progress", "done": False }, "type": "status" })
 		page = RedditPage(id, page_type).get_page()
 		await __event_emitter__({ "data": { "description": f"Retrieved {len(page.posts)} posts and {len(page.comments)} comments from {page_type.value}/{id}.", "status": "complete", "done": True }, "type": "status" })
-		return str(page)
-
-
-async def main():
-	__user__ = {
-		"valves": Tools.UserValves(
-			USER_AGENT="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36"
-		)
-	}
-	async def mock_event_emitter(event: dict): print("Event Emitted:", event["data"])
-	await Tools().get_reddit_feed("sysadmin", RedditPageType.SUBREDDIT, mock_event_emitter, __user__)
-	await Tools().get_reddit_feed("sysadmin/1eb73j6", RedditPageType.SUBREDDIT_COMMENTS, mock_event_emitter, __user__)
-	await Tools().get_reddit_feed("natfan", RedditPageType.USER, mock_event_emitter, __user__)
-
-import asyncio
-if __name__ == "__main__": asyncio.run(main())
\ No newline at end of file
+		return str(page)
\ No newline at end of file
diff --git a/test/reddit.test.py b/test/reddit.test.py
index 0bac032..42bff35 100644
--- a/test/reddit.test.py
+++ b/test/reddit.test.py
@@ -2,7 +2,7 @@
 import sys, os
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../src")))
 # Unfortunately, we don't get any type hinting, but this does work :)
-from reddit import Tools
+from reddit import RedditPageType, Tools
 
 async def main():
 	__user__ = {
@@ -11,8 +11,9 @@ async def main():
 		)
 	}
 	async def mock_event_emitter(event: dict): print("Event Emitted:", event["data"])
-	await Tools().get_subreddit_feed("sysadmin", mock_event_emitter, __user__)
-	await Tools().get_user_feed("NathanWindisch", mock_event_emitter, __user__)
+	await Tools().get_reddit_feed("sysadmin", RedditPageType.SUBREDDIT, mock_event_emitter, __user__)
+	await Tools().get_reddit_feed("sysadmin/1eb73j6", RedditPageType.SUBREDDIT_COMMENTS, mock_event_emitter, __user__)
+	await Tools().get_reddit_feed("natfan", RedditPageType.USER, mock_event_emitter, __user__)
 
 import asyncio
 if __name__ == "__main__": asyncio.run(main())
\ No newline at end of file