Added plotting of classified data