diff --git a/pyproject.toml b/pyproject.toml
index b490d4c..cd3dcf9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [build-system]
-requires = ["setuptools>=61.0"]
-build-backend = "setuptools.build_meta"
+requires = ["hatchling"]
+build-backend = "hatchling.build"
 
 [project]
 name = "drama_generator"
@@ -18,4 +18,10 @@ classifiers = [
 ]
 
 [project.urls]
-"Homepage" = "https://git.weber.codes/tom/drama_generator.git"
\ No newline at end of file
+"repository" = "https://git.weber.codes/tom/drama_generator.git"
+
+[tool.hatch.build]
+include = [
+  "drama_generator/data",
+    "drama_generator/saved_models",
+]
\ No newline at end of file