Updated with confusion matrix