diff --git a/README.md b/README.md
index ad50c7a..e931e8b 100644
--- a/README.md
+++ b/README.md
@@ -6,8 +6,8 @@ Simple node for setting the sigma values directly. Note, for a full denoise the
-# Shoutout
-[ComfyUI_tinyterraNodes](https://github.com/TinyTerra/ComfyUI_tinyterraNodes) - Based on code from this repo, which showed me how to hide and show widgets :)
+# Props
+ This repo is based on code from [ComfyUI_tinyterraNodes](https://github.com/TinyTerra/ComfyUI_tinyterraNodes), which showed me how to hide and show widgets :)
# My other scheduler node
-[ComfyUI-SchedulerMixer](https://github.com/BlakeOne/ComfyUI-SchedulerMixer) - Create a custom scheduler from a weighted average of the built-in schedulers
+[ComfyUI-SchedulerMixer](https://github.com/BlakeOne/ComfyUI-SchedulerMixer) - Create a custom scheduler from a weighted average of the built-in schedulers
\ No newline at end of file
diff --git a/__init__.py b/__init__.py
index 08bfbcf..eecab4f 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,28 +1,7 @@
-# Comfy mappings
from .node import CustomScheduler
NODE_CLASS_MAPPINGS = {"CustomScheduler": CustomScheduler}
NODE_DISPLAY_NAME_MAPPINGS = {"CustomScheduler" : "CustomScheduler"}
-__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+WEB_DIRECTORY = "./js"
-# Copy extension to comfy
-import folder_paths
-import shutil
-import os
-
-cwd_path = os.path.dirname(os.path.realpath(__file__))
-comfy_path = folder_paths.base_path
-web_extension_path = os.path.join(comfy_path, "web", "extensions", "CustomScheduler")
-widgets_js_file = os.path.join(cwd_path, "extension.js")
-
-if not os.path.exists(web_extension_path):
- os.makedirs(web_extension_path)
-else:
- try:
- shutil.rmtree(web_extension_path)
- os.makedirs(web_extension_path)
- except:
- print("Unable to remove old web extension.")
- pass
-
-shutil.copy(widgets_js_file, web_extension_path)
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
\ No newline at end of file
diff --git a/extension.js b/js/extension.js
similarity index 100%
rename from extension.js
rename to js/extension.js