diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..4f059a2
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,12 @@
+from .smart_resolution import SmartResolutionPicker
+from .smart_latent import SmartLatentGenerator
+
+NODE_CLASS_MAPPINGS = {
+    "SmartResolutionPicker": SmartResolutionPicker,
+    "SmartLatentGenerator": SmartLatentGenerator,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "SmartResolutionPicker": "Smart Resolution Picker",
+    "SmartLatentGenerator": "Smart Latent Generator",
+}