Adding a confusion matrix calculation feature