diff --git a/pyproject.toml b/pyproject.toml
index 7aa6d79..d305432 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -65,4 +65,4 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/ml_pipelines_kfp"]
\ No newline at end of file
+packages = ["src/ml_pipelines_kfp", "src/dataflow", "src/feature_store"]
\ No newline at end of file
diff --git a/src/dataflow/__init__.py b/src/dataflow/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/dataflow/models/__init__.py b/src/dataflow/models/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/src/dataflow/utils/__init__.py b/src/dataflow/utils/__init__.py
new file mode 100644
index 0000000..e69de29