diff --git a/README.md b/README.md
index 7ee363c..d895368 100644
--- a/README.md
+++ b/README.md
@@ -1 +1,2 @@
-# ComfyUI-Flux-TryOff
\ No newline at end of file
+# ComfyUI-Flux-TryOff
+
diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..9ffafaf
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,15 @@
+from .try_off_nodes import *
+
+# Node Registrations
+NODE_CLASS_MAPPINGS = {
+    "TryOffModelNode": TryOffModelNode,
+    "FluxFillModelNode": FluxFillModelNode,
+    "TryOffRunNode": TryOffRunNode,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "TryOffModelNode": "TryOff Model Loader",
+    "FluxFillModelNode": "FluxFill Model Loader",
+    "TryOffRunNode": "Run TryOff Inference",
+}
+
diff --git a/TryOffNodes.py b/try_off_nodes.py
similarity index 93%
rename from TryOffNodes.py
rename to try_off_nodes.py
index e0eecfb..ce71350 100644
--- a/TryOffNodes.py
+++ b/try_off_nodes.py
@@ -140,15 +140,3 @@ class TryOffRunNode:
         return garment_result, tryoff_result
 
 
-# Node Registrations
-NODE_CLASS_MAPPINGS = {
-    "TryOffModelNode": TryOffModelNode,
-    "FluxFillModelNode": FluxFillModelNode,
-    "TryOffRunNode": TryOffRunNode,
-}
-
-NODE_DISPLAY_NAME_MAPPINGS = {
-    "TryOffModelNode": "TryOff Model Loader",
-    "FluxFillModelNode": "FluxFill Model Loader",
-    "TryOffRunNode": "Run TryOff Inference",
-}