- Added method to get confusion matrix