class probabilities error corrected