diff --git a/python b/python
new file mode 100644
index 00000000..289a79d6
--- /dev/null
+++ b/python
@@ -0,0 +1,9 @@
+"""
+Cognition tasks for training reasoning capabilities:
+- Pattern recognition
+- Sequence completion
+- Logical reasoning
+- Working memory
+"""
+
+__all__ = []
diff --git a/reasoning_gym/__init__.py b/reasoning_gym/__init__.py
index c1b93a3a..aca8e226 100644
--- a/reasoning_gym/__init__.py
+++ b/reasoning_gym/__init__.py
@@ -3,6 +3,7 @@ Reasoning Gym - A library of procedural dataset generators for training reasonin
 """
 
 from . import arithmetic
+from . import cognition
 
 __version__ = "0.1.0"
-__all__ = ["arithmetic"]
+__all__ = ["arithmetic", "cognition"]