diff --git a/scripts/Confusion_Matrix/make_confusion_matrix.py b/scripts/Confusion_Matrix/make_confusion_matrix.py
index 2200f12..635e723 100644
--- a/scripts/Confusion_Matrix/make_confusion_matrix.py
+++ b/scripts/Confusion_Matrix/make_confusion_matrix.py
@@ -51,9 +51,4 @@ def make_confusion_matrix(y_true, y_pred, classes=None, figsize=(10, 10), text_s
 
   # Save the figure to the current working directory
   if savefig:
-    fig.savefig("confusion_matrix.png")
-
-y_true = [0, 0, 0, 0, 1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 1, 0]
-y_pred = [0, 0, 1, 0, 1, 1, 0, 0, 1, 1, 0, 0, 1, 0, 1, 1, 1, 0, 1, 0]
-
-make_confusion_matrix(y_true, y_pred)
\ No newline at end of file
+    fig.savefig("confusion_matrix.png")
\ No newline at end of file