diff --git a/cognee-mcp/src/cognee_client.py b/cognee-mcp/src/cognee_client.py
index 9d98cb0b5..3ffbca8d8 100644
--- a/cognee-mcp/src/cognee_client.py
+++ b/cognee-mcp/src/cognee_client.py
@@ -192,9 +192,7 @@ class CogneeClient:
 
             with redirect_stdout(sys.stderr):
                 results = await self.cognee.search(
-                    query_type=SearchType[query_type.upper()],
-                    query_text=query_text,
-                    top_k=top_k
+                    query_type=SearchType[query_type.upper()], query_text=query_text, top_k=top_k
                 )
                 return results
 
diff --git a/run_cognee.py b/run_cognee.py
new file mode 100644
index 000000000..206c5fc4d
--- /dev/null
+++ b/run_cognee.py
@@ -0,0 +1,24 @@
+import asyncio
+import cognee
+
+DATASET = "sheet6_demo"
+
+
+async def main():
+    # optional clean slate
+    await cognee.prune.prune_data()
+    await cognee.prune.prune_system(metadata=True)
+
+    await cognee.add("Cognee turns documents into AI memory.", dataset_name=DATASET)
+
+    # IMPORTANT: use datasets=[...] (not dataset_name=...)
+    await cognee.cognify(datasets=[DATASET])
+
+    results = await cognee.search(query_text="What does Cognee do?", datasets=[DATASET])
+
+    for r in results:
+        print(r.get("search_result"))
+
+
+if __name__ == "__main__":
+    asyncio.run(main())