diff --git a/python/stackforge/__init__.py b/python/stackforge/__init__.py
index 485f44a..6d2b678 100644
--- a/python/stackforge/__init__.py
+++ b/python/stackforge/__init__.py
@@ -1 +1,6 @@
 __version__ = "0.1.1"
+
+# Re-export all public classes from the Rust extension
+from stackforge.stackforge import LayerIndex, LayerKind, Packet
+
+__all__ = ["Packet", "LayerKind", "LayerIndex", "__version__"]