diff --git a/reasoning_gym/algorithmic/__init__.py b/reasoning_gym/algorithmic/__init__.py
index 4c7c7d6c..41e12f67 100644
--- a/reasoning_gym/algorithmic/__init__.py
+++ b/reasoning_gym/algorithmic/__init__.py
@@ -6,15 +6,19 @@ Algorithmic tasks for training reasoning capabilities:
 - Pattern matching
 """
 
+from reasoning_gym.arithmetic.basic_arithmetic import arithmetic_dataset
+from reasoning_gym.arithmetic.chain_sum import chain_sum
 from .base_conversion import BaseConversionConfig, BaseConversionDataset, base_conversion_dataset
 from .letter_counting import LetterCountingConfig, LetterCountingDataset, letter_counting_dataset
 from .number_filtering import NumberFilteringConfig, NumberFilteringDataset, number_filtering_dataset
 from .word_reversal import WordReversalConfig, WordReversalDataset, word_reversal_dataset
 
 __all__ = [
+    "arithmetic_dataset",
     "BaseConversionConfig",
     "BaseConversionDataset",
     "base_conversion_dataset",
+    "chain_sum",
     "LetterCountingConfig", 
     "LetterCountingDataset", 
     "letter_counting_dataset",