Added new plot as a diagnostic