Added diagnostic plots and examples