diff --git a/__init__.py b/__init__.py
index 5a3921a..221974d 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilitySearchAndRecolor, StabilityControlStyle, StabilityFastUpscale, StabilityReplaceRelight
+from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilityFastUpscale, StabilityControlStyle, StabilitySearchAndRecolor
 
 NODE_CLASS_MAPPINGS = {
     "Stability Conservative Upscale": StabilityConservativeUpscale,
@@ -13,8 +13,7 @@ NODE_CLASS_MAPPINGS = {
     "Stability SD3": StabilitySD3,
     "Stability Control Sketch": StabilityControlSketch,
     "Stability Control Structure": StabilityControlStructure,
-    "Stability Search And Recolor": StabilitySearchAndRecolor,
-    "Stability Control Style": StabilityControlStyle,
     "Stability Fast Upscale": StabilityFastUpscale,
-    "Stability Replace Background and Relight": StabilityReplaceRelight,
-}
+    "Stability Style": StabilityControlStyle,
+    "Stability Search and Recolor": StabilitySearchAndRecolor,
+}
\ No newline at end of file
diff --git a/stability_api.py b/stability_api.py
index 9eb4760..155fb7e 100644
--- a/stability_api.py
+++ b/stability_api.py
@@ -267,6 +267,10 @@ class StabilityRemoveBackground(StabilityBase):
         "required": {
             "image": ("IMAGE",),
         },
+        "optional": {
+            "output_format": (["png", "webp"],),
+            "api_key_override": ("STRING", {"multiline": False}),
+        }
     }
 
 
@@ -385,7 +389,7 @@ class StabilityOutpainting(StabilityBase):
             "left": ("INT", {"default": 0, "min": 0, "max": 512}),
             "right": ("INT", {"default": 0, "min": 0, "max": 512}),
             "up": ("INT", {"default": 0, "min": 0, "max": 512}),
-            "down": ("INT", {"default": 0, "min": 0, "max": 512}),\
+            "down": ("INT", {"default": 0, "min": 0, "max": 512}),
         },
         "optional": {
             "prompt": ("STRING", {"multiline": True}),
@@ -431,26 +435,19 @@ class StabilityControlStructure(StabilityBase):
         },
     }
 
-
-class StabilitySearchAndRecolor(StabilityBase):
-    API_ENDPOINT = "stable-image/edit/search-and-recolor"
+class StabilityFastUpscale(StabilityBase):
+    API_ENDPOINT = "stable-image/upscale/fast"
     ACCEPT = "image/*"
     INPUT_SPEC = {
         "required": {
             "image": ("IMAGE",),
-            "select_prompt": ("STRING", {"multiline": True}),
-            "prompt": ("STRING", {"multiline": True}),
         },
         "optional": {
-            "negative_prompt": ("STRING", {"multiline": True}),
-            "grow_mask": ("INT", {"default": 3, "min": 0, "max": 20, "step": 1}),
-            "seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
-            "api_key_override": ("STRING", {"multiline": False}),
             "output_format": (["png", "webp", "jpeg"],),
-        },
+            "api_key_override": ("STRING", {"multiline": False}),
+        }
     }
 
-
 class StabilityControlStyle(StabilityBase):
     API_ENDPOINT = "stable-image/control/style"
     ACCEPT = "image/*"
@@ -469,18 +466,22 @@ class StabilityControlStyle(StabilityBase):
         },
     }
 
-
-class StabilityFastUpscale(StabilityBase):
-    API_ENDPOINT = "stable-image/upscale/fast"
+class StabilitySearchAndRecolor(StabilityBase):
+    API_ENDPOINT = "stable-image/edit/search-and-recolor"
     ACCEPT = "image/*"
     INPUT_SPEC = {
         "required": {
             "image": ("IMAGE",),
+            "select_prompt": ("STRING", {"multiline": True}),
+            "prompt": ("STRING", {"multiline": True}),
         },
         "optional": {
-            "output_format": (["png", "webp", "jpeg"],),
+            "negative_prompt": ("STRING", {"multiline": True}),
+            "grow_mask": ("INT", {"default": 3, "min": 0, "max": 20, "step": 1}),
+            "seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
             "api_key_override": ("STRING", {"multiline": False}),
-        }
+            "output_format": (["png", "webp", "jpeg"],),
+        },
     }
 
 class StabilityReplaceRelight(StabilityBase):