diff --git a/README.md b/README.md
index 80b00849..ee288ce6 100644
--- a/README.md
+++ b/README.md
@@ -31,7 +31,7 @@ The goal is to generate virtually infinite data with adjustable complexity.
 #### Logic Tasks
 - `PropositionalLogicDataset`: Generate propositional logic reasoning problems
 
-#### Relationship Tasks
+#### Graph Tasks
 - `FamilyRelationshipsDataset`: Generate family relationship reasoning tasks with family trees
 
 #### Game Tasks
diff --git a/reasoning_gym/__init__.py b/reasoning_gym/__init__.py
index 0d3f0e9e..52741f8c 100644
--- a/reasoning_gym/__init__.py
+++ b/reasoning_gym/__init__.py
@@ -2,7 +2,7 @@
 Reasoning Gym - A library of procedural dataset generators for training reasoning models
 """
 
-from . import algorithmic, algebra, arithmetic, cognition, data, games, logic, relationships
+from . import algorithmic, algebra, arithmetic, cognition, data, games, graphs, logic
 
 __version__ = "0.1.1"
-__all__ = ["arithmetic", "algorithmic", "algebra", "cognition", "data", "games", "logic", "relationships"]
+__all__ = ["arithmetic", "algorithmic", "algebra", "cognition", "data", "games", "graphs", "logic"]