Add code to create a confusion matrix