Added the confusion matrix part.