diff --git a/README.md b/README.md
index ed15e319..4b91500c 100644
--- a/README.md
+++ b/README.md
@@ -7,7 +7,7 @@ The goal is to generate virtually infinite data with adjustable complexity.
 ### Task Overview
 
 #### Arithmetic Tasks
-- `ArithmeticDataset`: Generate arithmetic expressions with configurable complexity and operators (+, -, *)
+- `BasicArithmeticDataset`: Generate arithmetic expressions with configurable complexity and operators (+, -, *)
 - `ChainSum`: Generate addition/subtraction chains with configurable length and digit counts
 - `FractionSimplificationDataset`: Generate fraction simplification tasks with configurable complexity
 - `GCDDataset`: Generate Greatest Common Divisor problems with configurable number of integers
diff --git a/reasoning_gym/arithmetic/__init__.py b/reasoning_gym/arithmetic/__init__.py
index c1614f99..c7953db8 100644
--- a/reasoning_gym/arithmetic/__init__.py
+++ b/reasoning_gym/arithmetic/__init__.py
@@ -6,7 +6,7 @@ Arithmetic tasks for training reasoning capabilities:
 - Leg counting
 """
 
-from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig, arithmetic_dataset
+from .basic_arithmetic import BasicArithmeticDataset, ArithmeticDatasetConfig, basic_arithmetic_dataset
 from .chain_sum import ChainSum, ChainSumConfig, chain_sum_dataset
 from .fraction_simplification import FractionSimplificationConfig, FractionSimplificationDataset, fraction_simplification_dataset
 from .gcd import GCDConfig, GCDDataset, gcd_dataset
@@ -15,9 +15,9 @@ from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_da
 from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset, prime_factorization_dataset
 
 __all__ = [
-    "ArithmeticDataset",
+    "BasicArithmeticDataset",
     "ArithmeticDatasetConfig",
-    "arithmetic_dataset",
+    "basic_arithmetic_dataset",
     "ChainSum",
     "ChainSumConfig",
     "chain_sum_dataset",
diff --git a/reasoning_gym/arithmetic/basic_arithmetic.py b/reasoning_gym/arithmetic/basic_arithmetic.py
index cde9a543..80a45edb 100644
--- a/reasoning_gym/arithmetic/basic_arithmetic.py
+++ b/reasoning_gym/arithmetic/basic_arithmetic.py
@@ -31,8 +31,8 @@ class ArithmeticDatasetConfig:
             assert op in ["+", "-", "*"], f"unsupported operator: {op}"
 
 
-class ArithmeticDataset(ProceduralDataset):
-    """Dataset that generates arithmetic tasks with configurable complexity"""
+class BasicArithmeticDataset(ProceduralDataset):
+    """Dataset that generates basic arithmetic tasks with configurable complexity"""
 
     def __init__(self, config: ArithmeticDatasetConfig):
         self.config = config
@@ -155,7 +155,7 @@ class ArithmeticDataset(ProceduralDataset):
             return rng.choice(templates).format(expression)
 
 
-def arithmetic_dataset(
+def basic_arithmetic_dataset(
     min_terms: int = 2,
     max_terms: int = 6,
     min_digits: int = 1,
@@ -167,8 +167,8 @@ def arithmetic_dataset(
     size: int = 500,
     format_style: Literal["simple", "natural"] = "simple",
     whitespace: Literal["no_space", "single", "random"] = "single",
-) -> ArithmeticDataset:
-    """Create an ArithmeticDataset with the given configuration.
+) -> BasicArithmeticDataset:
+    """Create a BasicArithmeticDataset with the given configuration.
 
     Args:
         min_terms: Minimum number of terms in expressions
@@ -198,4 +198,4 @@ def arithmetic_dataset(
         format_style=format_style,
         whitespace=whitespace,
     )
-    return ArithmeticDataset(config)
+    return BasicArithmeticDataset(config)