added diagnostic plot wrapper