Added additional diagnostic plots