updates for diagnostic plots

addressing