   axs[0][0].title.set_text("Image")
   axs[0][0].set(xlabel: "x displacement", ylabel: "y displacement")
 
-  let fg_m = axs[0][1].imshow(fg.makeNumpyArray(), cmap: "hot", interpolation: "nearest") // , vmin: trans_min, vmax: trans_max)
+  // Plot the foreground model
+  let fg_m = normalizeScale ?
+    axs[0][1].imshow(fg.makeNumpyArray(), cmap: "hot", interpolation: "nearest", vmin: trans_min, vmax: trans_max) :
+    axs[0][1].imshow(fg.makeNumpyArray(), cmap: "hot", interpolation: "nearest")
+
   fig.colorbar(fg_m, ax: axs[0][1])
   axs[0][1].title.set_text("Foreground Response")
   axs[0][1].set(xlabel: "x displacement", ylabel: "y displacement")
 
-  let bg_m = axs[1][0].imshow(bg.makeNumpyArray(), cmap: "hot", interpolation: "nearest" ) // , vmin: trans_min, vmax: trans_max)
+  // Plot the background model
+  let bg_m = normalizeScale ?
+    axs[1][0].imshow(bg.makeNumpyArray(), cmap: "hot", interpolation: "nearest", vmin: trans_min, vmax: trans_max):
+    axs[1][0].imshow(bg.makeNumpyArray(), cmap: "hot", interpolation: "nearest")
+
   fig.colorbar(bg_m, ax: axs[1][0])
   axs[1][0].title.set_text("Background Response")
   axs[1][0].set(xlabel: "x displacement", ylabel: "y displacement")
 
-  let pcm = axs[1][1].imshow(e.makeNumpyArray(), cmap: "hot", interpolation: "nearest") // , vmin: trans_min, vmax: trans_max)
+  // Plot the response (error)
+  let pcm = normalizeScale ?
+    axs[1][1].imshow(e.makeNumpyArray(), cmap: "hot", interpolation: "nearest", vmin: trans_min, vmax: trans_max):
+    axs[1][1].imshow(e.makeNumpyArray(), cmap: "hot", interpolation: "nearest")
+
   fig.colorbar(pcm, ax: axs[1][1])
   axs[1][1].title.set_text("Total Response")
   axs[1][1].set(xlabel: "x displacement", ylabel: "y displacement")

  statistics: FrameStatistics,
  encoder: Encoder,
  foregroundModel: FGModel,
  backgroundModel: BGModel
) -> (PythonObject, PythonObject) {
  let plt = Python.import("matplotlib.pyplot")
  let (fg, bg, e) = errorPlaneTranslation(

  return (fig, ax)
}

/// Calculate the translation error plane (X-Y)
public func errorPlaneTranslation<
  Encoder: AppearanceModelEncoder,
  FGModel: GenerativeDensity,
