classifier now returns the probability for each class
old class removed