diff --git a/pyinfra/__init__.py b/pyinfra/__init__.py
index e69de29..9d92b43 100644
--- a/pyinfra/__init__.py
+++ b/pyinfra/__init__.py
@@ -0,0 +1,3 @@
+from pyinfra import k8s_probes, queue, storage, config
+
+__all__ = ["k8s_probes", "queue", "storage", "config"]
\ No newline at end of file
diff --git a/pyinfra/k8s_probes/__init__.py b/pyinfra/k8s_probes/__init__.py
index e69de29..ecc33ff 100644
--- a/pyinfra/k8s_probes/__init__.py
+++ b/pyinfra/k8s_probes/__init__.py
@@ -0,0 +1,3 @@
+from pyinfra.k8s_probes import startup
+
+__all__ = ["startup"]
\ No newline at end of file
diff --git a/pyinfra/queue/__init__.py b/pyinfra/queue/__init__.py
index e69de29..c980a1e 100644
--- a/pyinfra/queue/__init__.py
+++ b/pyinfra/queue/__init__.py
@@ -0,0 +1,3 @@
+from pyinfra.queue import queue_manager
+
+__all__ = ["queue_manager"]
\ No newline at end of file
diff --git a/pyinfra/storage/__init__.py b/pyinfra/storage/__init__.py
index e69de29..f01d6da 100644
--- a/pyinfra/storage/__init__.py
+++ b/pyinfra/storage/__init__.py
@@ -0,0 +1,4 @@
+from pyinfra.storage import adapters, storage
+from pyinfra.storage.storage import get_storage
+
+__all__ = ["adapters", "storage"]
\ No newline at end of file