diff --git a/README.md b/README.md
index 88b0cda..6193c32 100644
--- a/README.md
+++ b/README.md
@@ -22,7 +22,8 @@ image_layering:
 color_correction:
 - Adjusts the color of the target image according to another image; ported from stable diffusion WebUI
 
-
+model_router:
+- Batch reroutes `model`, `clip`, `vae`, `condition_1` and `condition_2` for cleaner workflow
 
 ## External Nodes
 
diff --git a/__init__.py b/__init__.py
index e2cbabe..c153482 100644
--- a/__init__.py
+++ b/__init__.py
@@ -5,8 +5,8 @@ import custom_nodes.comfy_nodes_trojblue.model_router as model_router
 
 
 NODE_CLASS_MAPPINGS = {
-    "layering": image_layering.Layering,                      # Layering
-    "color_correction": color_correction.ColorCorrectionNode,  # ColorCorrectionNode
+    "trLayering": image_layering.Layering,                      # Layering
+    "trColorCorrection": color_correction.ColorCorrectionNode,  # ColorCorrectionNode
     "trRouter": model_router.ModelRouterPlugin,          # ModelRouterPlugin
 
 }
diff --git a/color_correction.py b/color_correction.py
index d79613e..5fd5657 100644
--- a/color_correction.py
+++ b/color_correction.py
@@ -57,6 +57,3 @@ class ColorCorrectionNode:
 
         return (out_image,)
 
-NODE_CLASS_MAPPINGS = {
-    "ColorCorrectionNode": ColorCorrectionNode
-}
diff --git a/image_layering.py b/image_layering.py
index 0d7784d..2bddef3 100644
--- a/image_layering.py
+++ b/image_layering.py
@@ -90,9 +90,6 @@ class Layering:
 
         return (out_image,)
 
-NODE_CLASS_MAPPINGS = {
-    "Layering": Layering,
-}
 
 
 
diff --git a/model_router.py b/model_router.py
index 27b552e..7f7468f 100644
--- a/model_router.py
+++ b/model_router.py
@@ -43,9 +43,3 @@ class ModelRouterPlugin:
     def execute(self, model=None, clip=None, vae=None, conditioning1=None, conditioning2=None):
         return model, clip, vae, conditioning1, conditioning2
 
-
-# A dictionary that contains all nodes you want to export with their names
-# NOTE: names should be globally unique
-NODE_CLASS_MAPPINGS = {
-    "trRouter": ModelRouterPlugin
-}