Create confusion_matrix.py