Updates to plotting and diagnostics.