diff --git a/fedmind/__init__.py b/fedmind/__init__.py
index e69de29..b86a5f4 100644
--- a/fedmind/__init__.py
+++ b/fedmind/__init__.py
@@ -0,0 +1,11 @@
+from .algs.fedavg import FedAvg
+from .server import FedAlg
+from .utils import EasyDict, StateDict
+
+
+__all__ = [
+    "FedAvg",
+    "FedAlg",
+    "EasyDict",
+    "StateDict",
+]