Really Minor: Maybe move figure/axes definitions down with the rest of the plotting code (i.e. below the function defs). Also, I think the inline comments should be removed here. (Maybe `ax1` -> `ax_f`, `ax2` -> `ax_c`?)