Modified the decision tree to print out diagnostic plots.