diff --git a/pyproject.toml b/pyproject.toml
index 6966c146..1dc6a3cc 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,6 +21,14 @@ classifiers = [
 "Homepage" = "https://github.com/open-thought/reasoning-gym"
 "Bug Tracker" = "https://github.com/open-thought/reasoning-gym/issues"
 
+[tool.hatch.build.targets.wheel]
+packages = ["reasoning_gym"]
+
+[tool.hatch.build]
+include = [
+  "reasoning_gym/**/*.txt",
+]
+
 [tool.black]
 line-length = 120
 target-version = ['py312']
diff --git a/reasoning_gym/data/__init__.py b/reasoning_gym/data/__init__.py
new file mode 100644
index 00000000..435fcb07
--- /dev/null
+++ b/reasoning_gym/data/__init__.py
@@ -0,0 +1 @@
+"""Package containing data files used by reasoning_gym"""