Added a legend to the diagnostic plot.
Notebook reflects inclusion of lagend in diagnostic plots.
Notebook reflects inclusion of lagend in diagnostic plots.