Update confusion_matrix.py