Fixed the confusion matrix