diff --git a/__init__.py b/__init__.py
new file mode 100755
index 0000000..1ce08df
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,13 @@
+import DetailModelWrapperNode from Detail_Model_Patch_Custom
+import DetailModelUnwrapperNode from Detail_Model_Patch_Custom
+
+# Node registration for ComfyUI
+NODE_CLASS_MAPPINGS = {
+    "DetailModelWrapperNode": DetailModelWrapperNode,
+    "DetailModelUnwrapperNode": DetailModelUnwrapperNode
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "DetailModelWrapperNode": "Detail Model Wrapper",
+    "DetailModelUnwrapperNode": "Detail Model Unwrapper"
+}