diff --git a/GALLERY.md b/GALLERY.md
index c99fb91a..07642aa7 100644
--- a/GALLERY.md
+++ b/GALLERY.md
@@ -1626,16 +1626,19 @@ Example tasks:
 ````
 Example 1:
 Question: Keep all numbers larger than -90 in this list: ['-95.00', '-51.0', '47.2942', '-82.612']
+Return the new list in the same format.
 Answer: ['-51.0', '47.2942', '-82.612']
 Metadata: {'original_numbers': ['-95.00', '-51.0', '47.2942', '-82.612'], 'filter_value': '-90', 'operation': 'keep_larger', 'result': ['-51.0', '47.2942', '-82.612']}
 
 Example 2:
 Question: Remove all numbers larger than 18.236 in this list: ['-42.8', '91.88', '34']
+Return the new list in the same format.
 Answer: ['-42.8']
 Metadata: {'original_numbers': ['-42.8', '91.88', '34'], 'filter_value': '18.236', 'operation': 'remove_larger', 'result': ['-42.8']}
 
 Example 3:
 Question: Keep all numbers larger than 19.8962 in this list: ['4', '-64.7', '-42.1', '-77', '-79.9640', '37.76', '38.702', '18.20', '-28.34']
+Return the new list in the same format.
 Answer: ['37.76', '38.702']
 Metadata: {'original_numbers': ['4', '-64.7', '-42.1', '-77', '-79.9640', '37.76', '38.702', '18.20', '-28.34'], 'filter_value': '19.8962', 'operation': 'keep_larger', 'result': ['37.76', '38.702']}
 
@@ -1809,7 +1812,7 @@ min_degree = 1
 max_degree = 3
 min_polynomials = 2
 max_polynomials = 3
-single_variable = (True,)
+single_variable = True
 operators = ('+', '-')
 seed = 42
 size = 500
@@ -1820,17 +1823,17 @@ Example tasks:
 Example 1:
 Question: Calculate the following: (65*x - 72)*(105*x - 125)
 Answer: 6825*x**2 - 15685*x + 9000
-Metadata: {'polynomial_expr': '(65*x - 72)*(105*x - 125)', 'single_variable': (True,), 'result': '6825*x**2 - 15685*x + 9000'}
+Metadata: {'polynomial_expr': '(65*x - 72)*(105*x - 125)', 'single_variable': True, 'result': '6825*x**2 - 15685*x + 9000'}
 
 Example 2:
 Question: Calculate the following: (-9*x**2 - 28*x)*(86*x**2 - 2*x - 13)
 Answer: -774*x**4 - 2390*x**3 + 173*x**2 + 364*x
-Metadata: {'polynomial_expr': '(-9*x**2 - 28*x)*(86*x**2 - 2*x - 13)', 'single_variable': (True,), 'result': '-774*x**4 - 2390*x**3 + 173*x**2 + 364*x'}
+Metadata: {'polynomial_expr': '(-9*x**2 - 28*x)*(86*x**2 - 2*x - 13)', 'single_variable': True, 'result': '-774*x**4 - 2390*x**3 + 173*x**2 + 364*x'}
 
 Example 3:
 Question: Calculate the following: (43 - 91*x)*(3*x**2 - 10*x)*(71*x**3 - 2*x - 29)
 Answer: -19383*x**6 + 73769*x**5 - 29984*x**4 + 5839*x**3 - 29271*x**2 + 12470*x
-Metadata: {'polynomial_expr': '(43 - 91*x)*(3*x**2 - 10*x)*(71*x**3 - 2*x - 29)', 'single_variable': (True,), 'result': '-19383*x**6 + 73769*x**5 - 29984*x**4 + 5839*x**3 - 29271*x**2 + 12470*x'}
+Metadata: {'polynomial_expr': '(43 - 91*x)*(3*x**2 - 10*x)*(71*x**3 - 2*x - 29)', 'single_variable': True, 'result': '-19383*x**6 + 73769*x**5 - 29984*x**4 + 5839*x**3 - 29271*x**2 + 12470*x'}
 
 ````
 
diff --git a/reasoning_gym/arc/__init__.py b/reasoning_gym/arc/__init__.py
index e1ca90b8..e9bf7086 100644
--- a/reasoning_gym/arc/__init__.py
+++ b/reasoning_gym/arc/__init__.py
@@ -1,3 +1,4 @@
+from .arc_1d import Arc1DConfig, Arc1DDataset
 from .rearc import ReArcConfig, ReArcDataset
 
-__all__ = ["ReArcDataset", "ReArcConfig"]
+__all__ = ["Arc1DConfig", "Arc1DDataset", "ReArcDataset", "ReArcConfig"]
diff --git a/reasoning_gym/cognition/arc_1d.py b/reasoning_gym/arc/arc_1d.py
similarity index 100%
rename from reasoning_gym/cognition/arc_1d.py
rename to reasoning_gym/arc/arc_1d.py
diff --git a/reasoning_gym/cognition/arc_1d_tasks.py b/reasoning_gym/arc/arc_1d_tasks.py
similarity index 100%
rename from reasoning_gym/cognition/arc_1d_tasks.py
rename to reasoning_gym/arc/arc_1d_tasks.py
diff --git a/reasoning_gym/cognition/__init__.py b/reasoning_gym/cognition/__init__.py
index 751c9ab8..5c0b7f8b 100644
--- a/reasoning_gym/cognition/__init__.py
+++ b/reasoning_gym/cognition/__init__.py
@@ -2,15 +2,12 @@
 Cognition tasks for training reasoning capabilities.
 """
 
-from .arc_1d import Arc1DConfig, Arc1DDataset
 from .color_cube_rotation import ColorCubeRotationConfig, ColorCubeRotationDataset
 from .figlet_fonts import FigletFontConfig, FigletFontDataset
 from .number_sequences import NumberSequenceConfig, NumberSequenceDataset
 from .rubiks_cube import RubiksCubeConfig, RubiksCubeDataset
 
 __all__ = [
-    "Arc1DConfig",
-    "Arc1DDataset",
     "ColorCubeRotationConfig",
     "ColorCubeRotationDataset",
     "FigletFontConfig",
diff --git a/tests/test_arc_1d.py b/tests/test_arc_1d.py
index 1679fb50..6c32f79e 100644
--- a/tests/test_arc_1d.py
+++ b/tests/test_arc_1d.py
@@ -1,6 +1,6 @@
 import pytest
 
-from reasoning_gym.cognition import Arc1DConfig, Arc1DDataset
+from reasoning_gym.arc import Arc1DConfig, Arc1DDataset
 
 
 def test_arc_1d_config_validation():
diff --git a/tests/test_arc_1d_tasks.py b/tests/test_arc_1d_tasks.py
index 98d9d2ef..4692b030 100644
--- a/tests/test_arc_1d_tasks.py
+++ b/tests/test_arc_1d_tasks.py
@@ -2,7 +2,7 @@ import random
 
 import pytest
 
-from reasoning_gym.cognition.arc_1d_tasks import (
+from reasoning_gym.arc.arc_1d_tasks import (
     task_block_and_noise_remove,
     task_block_and_noise_remove_inside,
     task_block_scale_to_dot,