diff --git a/extensions/colors.js b/extensions/colors.js
index 706760d..3089e80 100644
--- a/extensions/colors.js
+++ b/extensions/colors.js
@@ -9,8 +9,8 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
-import { $el } from "../../../scripts/ui.js";
+import { app } from "../scripts/app.js";
+import { $el } from "../scripts/ui.js";
 
 function getColor(index, lengthOfItems, l = 0.5) {
   // Normalize the index value to be between 0 and 360 for full spectrum of hue
@@ -154,28 +154,28 @@ function setColorMode(value, app) {
     case 1:
       app.graph.afterChange = function () {
         uncolor(app);
-        return applyChanges?.apply(this, arguments);
+        return afterChange?.apply(this, arguments);
       };
       uncolor(app);
       break;
     case 2:
       app.graph.afterChange = function () {
         rainbowify(app);
-        return applyChanges?.apply(this, arguments);
+        return afterChange?.apply(this, arguments);
       };
       rainbowify(app);
       break;
     case 3:
       app.graph.afterChange = function () {
         colorByType(app);
-        return applyChanges?.apply(this, arguments);
+        return afterChange?.apply(this, arguments);
       };
       colorByType(app);
       break;
     case 4:
       app.graph.afterChange = function () {
         colorPositiveNegative(app);
-        return applyChanges?.apply(this, arguments);
+        return afterChange?.apply(this, arguments);
       };
       colorPositiveNegative(app);
       break;
diff --git a/extensions/connectionsWidth.js b/extensions/connectionsWidth.js
index 64a0870..1fadf7e 100644
--- a/extensions/connectionsWidth.js
+++ b/extensions/connectionsWidth.js
@@ -9,7 +9,7 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
+import { app } from "../scripts/app.js";
 
 /**
  * Connections Width
diff --git a/extensions/forceBoxNodes.js b/extensions/forceBoxNodes.js
index 3992d20..b94c000 100644
--- a/extensions/forceBoxNodes.js
+++ b/extensions/forceBoxNodes.js
@@ -9,7 +9,7 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
+import { app } from "../scripts/app.js";
 
 /**
  * Render all nodes as box
diff --git a/extensions/forceSnapToGrid.js b/extensions/forceSnapToGrid.js
index 2509933..8a876b2 100644
--- a/extensions/forceSnapToGrid.js
+++ b/extensions/forceSnapToGrid.js
@@ -9,7 +9,7 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
+import { app } from "../scripts/app.js";
 
 /**
  * Always snap to grid
diff --git a/extensions/linksRenderMode.js b/extensions/linksRenderMode.js
index 895c809..1feef82 100644
--- a/extensions/linksRenderMode.js
+++ b/extensions/linksRenderMode.js
@@ -10,8 +10,8 @@
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
 
-import { app } from "../../../scripts/app.js";
-import { $el } from "../../../scripts/ui.js";
+import { app } from "../scripts/app.js";
+import { $el } from "../scripts/ui.js";
 
 /**
  * Toggle links render mode
diff --git a/extensions/nodes.js b/extensions/nodes.js
index 6aae85e..73cc501 100644
--- a/extensions/nodes.js
+++ b/extensions/nodes.js
@@ -9,7 +9,7 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
+import { app } from "../scripts/app.js";
 
 /**
  * Nodes
@@ -33,14 +33,12 @@ app.registerExtension({
     });
   },
   async setup(app) {
-    console.log(app);
     app.graph._nodes.forEach((node) => {
       let active = JSON.parse(
         window.localStorage.getItem(`Comfy.Settings.${nodesName}`) ?? "false",
       );
       const onResize = node.onResize;
       node.onResize = function (size) {
-        console.log(active);
         if (active) {
           Array.from(store.stack)
             .filter((node_) => node_ !== node)
diff --git a/extensions/pinNodes.js b/extensions/pinNodes.js
index f915d2c..f82a02f 100644
--- a/extensions/pinNodes.js
+++ b/extensions/pinNodes.js
@@ -9,8 +9,8 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
-import { $el } from "../../../scripts/ui.js";
+import { app } from "../scripts/app.js";
+import { $el } from "../scripts/ui.js";
 
 /**
  * Pin/Unpin all nodes on canvas
diff --git a/extensions/renderShadows.js b/extensions/renderShadows.js
index 39b2285..d5bd869 100644
--- a/extensions/renderShadows.js
+++ b/extensions/renderShadows.js
@@ -9,7 +9,7 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
+import { app } from "../scripts/app.js";
 
 /**
  * Render Shadow
diff --git a/extensions/textFields.js b/extensions/textFields.js
index 9bdbf86..2944fff 100644
--- a/extensions/textFields.js
+++ b/extensions/textFields.js
@@ -9,8 +9,8 @@
  * GitHub: https://github.com/failfa-st
  * Discord: https://discord.com/invite/m3TBB9XEkb
  */
-import { app } from "../../../scripts/app.js";
-import { $el } from "../../../scripts/ui.js";
+import { app } from "../scripts/app.js";
+import { $el } from "../scripts/ui.js";
 
 /**
  * Font size of prompt fields
diff --git a/scripts/app.js b/scripts/app.js
new file mode 100644
index 0000000..342dd3b
--- /dev/null
+++ b/scripts/app.js
@@ -0,0 +1 @@
+export *  from "../../../scripts/app.js";
diff --git a/scripts/ui.js b/scripts/ui.js
new file mode 100644
index 0000000..ff3c8f7
--- /dev/null
+++ b/scripts/ui.js
@@ -0,0 +1 @@
+export * from "../../../scripts/ui.js";