diff --git a/figures/combine.png b/figures/combine.png
index ca86a8e..8debc33 100644
Binary files a/figures/combine.png and b/figures/combine.png differ
diff --git a/figures/deepxtrace.png b/figures/deepxtrace.png
index 8e047fd..757c309 100644
Binary files a/figures/deepxtrace.png and b/figures/deepxtrace.png differ
diff --git a/figures/dispatch.png b/figures/dispatch.png
index 1f974b6..757c309 100644
Binary files a/figures/dispatch.png and b/figures/dispatch.png differ
diff --git a/tools/deepxtrace_heatmap.py b/tools/deepxtrace_heatmap.py
index 3cf5ed2..440debb 100644
--- a/tools/deepxtrace_heatmap.py
+++ b/tools/deepxtrace_heatmap.py
@@ -11,12 +11,14 @@
 def create_optimized_ryg_cmap():
     """Create an optimized Red-Yellow-Green colormap (6-color scale)"""
     colors = [
-        (0.0, "#63BE7B"),  # Green
-        (0.3, "#A8D08D"),  # Green to Yellow-Green
-        (0.5, "#FFEB84"),  # Center Yellow
-        (0.7, "#F4B084"),  # Yellow to Orange
-        (0.9, "#FF7C7C"),  # Light Red
-        (1.0, "#C00000")   # Red
+        (0.00, "#4CAF50"),   # Green
+        (0.15, "#81C784"),   # Light Green
+        (0.30, "#AED581"),   # Green-Yellow
+        (0.45, "#FFF176"),   # Light Yellow
+        (0.55, "#FFD54F"),   # Yellow
+        (0.70, "#FFB74D"),   # Yellow-Orange
+        (0.85, "#FF8A65"),   # Light Red
+        (1.00, "#E53935")    # Red
     ]
     return LinearSegmentedColormap.from_list("optimized_ryg", colors)