added some code for confusion matrix