diff --git a/_mappings.py b/_mappings.py
index fd76c0d..df9a5c6 100644
--- a/_mappings.py
+++ b/_mappings.py
@@ -1,6 +1,8 @@
 from .generation import TWCUI_Util_GenerationParameters, TWCUI_Util_CommonSDXLResolutions
 from .generation import TWCUI_Util_GenerationPrompts
 
+from .image import TWCUI_Util_SaveImage
+
 from .util import TWCUI_Util_StringLiteral, TWCUI_Util_MultilineStringLiteral
 from .util import TWCUI_Util_IntLiteral, TWCUI_Util_FloatLiteral
 
@@ -10,6 +12,9 @@ NODE_CLASS_MAPPINGS = {
     "TWCUI_Util_GenerationParameters": TWCUI_Util_GenerationParameters,
     "TWCUI_Util_GenerationPrompts": TWCUI_Util_GenerationPrompts,
 
+    # Image category
+    "TWCUI_Util_SaveImage": TWCUI_Util_SaveImage,
+
     # Util category
     "TWCUI_Util_FloatLiteral": TWCUI_Util_FloatLiteral,
     "TWCUI_Util_IntLiteral": TWCUI_Util_IntLiteral,
@@ -23,6 +28,9 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "TWCUI_Util_GenerationParameters": "Generation Parameters",
     "TWCUI_Util_GenerationPrompts": "Prompts",
 
+    # Image category
+    "TWCUI_Util_SaveImage": "Save Image",
+
     # Util category
     "TWCUI_Util_FloatLiteral": "FLOAT Literal",
     "TWCUI_Util_IntLiteral": "INTEGER Literal",
diff --git a/saving.py b/image.py
similarity index 100%
rename from saving.py
rename to image.py