add more diagnostic plots