Updated with best confusion matrix