diff --git a/reasoning_gym/__init__.py b/reasoning_gym/__init__.py
index 43e69fd0..6cc3d50a 100644
--- a/reasoning_gym/__init__.py
+++ b/reasoning_gym/__init__.py
@@ -3,8 +3,10 @@ Reasoning Gym - A library of procedural dataset generators for training reasonin
 """
 
 from . import arithmetic
+from . import algorithmic
 from . import cognition
+from . import data
 from . import logic
 
 __version__ = "0.1.0"
-__all__ = ["arithmetic", "cognition", "logic"]
+__all__ = ["arithmetic", "algorithmic", "cognition", "data", "logic"]