diff --git a/BMM/BMM_Theory/bmm_theory/branches/__init__.py b/BMM/BMM_Theory/bmm_theory/branches/__init__.py
new file mode 100644
index 0000000..2c955b2
--- /dev/null
+++ b/BMM/BMM_Theory/bmm_theory/branches/__init__.py
@@ -0,0 +1,4 @@
+from .merge_batch import MergeBatchBranch
+from .iter_batch import IterBatchBranch
+
+__all__ = ["MergeBatchBranch", "IterBatchBranch"]