diff --git a/js/SimpleLoadLineFromTextFile.js b/js/SimpleLoadLineFromTextFile.js
index c5f9d28..225b357 100644
--- a/js/SimpleLoadLineFromTextFile.js
+++ b/js/SimpleLoadLineFromTextFile.js
@@ -5,25 +5,15 @@ app.registerExtension({
     name: "Simple Load Line From Text File",
     async beforeRegisterNodeDef(nodeType, nodeData, app) {
         if (nodeData.name === "Simple Load Line From Text File") {
-            const origOnNodeCreated = nodeType.prototype.onNodeCreated;
+            const originalOnNodeCreated = nodeType.prototype.onNodeCreated;
             nodeType.prototype.onNodeCreated = function() {
-                const r = origOnNodeCreated ? origOnNodeCreated.apply(this) : undefined;
+                if (originalOnNodeCreated) { originalOnNodeCreated.call(this); }
                 let counter = 0;
-                function find_count(w) { 
-                    return w.name === "count"
-                }
-                const count = this.widgets.find(find_count);
+                const count = this.widgets.find((w) => w.name === "count");
                 count.type = "converted-widget"; // hidden
-                function do_counter() { 
-                    return counter++; 
-                }
-                count.serializeValue = do_counter;
-                function reset_counter() { 
-                    counter = 0;
-                }
-                api.addEventListener("promptQueued", reset_counter); // reset
-                return r;
-            }
+                count.serializeValue = () => { return counter++; }
+                api.addEventListener("promptQueued", () => { counter = 0; }); // reset
+            };
         }
-    }
-})
+    },
+});
diff --git a/js/SimpleLoraLoader.js b/js/SimpleLoraLoader.js
new file mode 100644
index 0000000..9645abc
--- /dev/null
+++ b/js/SimpleLoraLoader.js
@@ -0,0 +1,18 @@
+import { app } from "/scripts/app.js";
+
+app.registerExtension({ 
+    name: "Simple Lora Loader", 
+    async beforeRegisterNodeDef(nodeType, nodeData) { 
+        if (nodeData.name === "Simple Lora Loader") { 
+            const originalOnNodeCreated = nodeType.prototype.onNodeCreated;
+            nodeType.prototype.onNodeCreated = function() {
+                if (originalOnNodeCreated) { originalOnNodeCreated.call(this); }
+                let counter = 1;
+                const count = this.widgets.find((w) => w.name === "count");
+                count.type = "converted-widget"; // hidden
+                count.serializeValue = () => { return counter++; }    
+                this.addWidget("button","Update LoRA dictionary",null,() => { counter = 0; },{ width: 150 });
+            };
+        }
+    },
+});
\ No newline at end of file