diff --git a/code/machine_learning_models/decision_tree.py b/code/machine_learning_models/decision_tree.py
index 1230948823560f20faad49d2328fc807c6f2b2ca..2d22b7679046cbbf10ff9b7a179fe4d61c8c1887 100644
--- a/code/machine_learning_models/decision_tree.py
+++ b/code/machine_learning_models/decision_tree.py
@@ -8,7 +8,7 @@ from sklearn.metrics import classification_report, confusion_matrix
 from sklearn.preprocessing import StandardScaler, LabelEncoder
 from sklearn.tree import DecisionTreeClassifier
 
-from utilities import plot_counts, plot_roc_curve
+from utilities import plot_counts
 from utilities import plot_features, ordinal_encode, normalize, plot_confusion_matrix, print_high_confidence_samples, import_data
 
 warnings.filterwarnings("ignore")
diff --git a/code/machine_learning_models/knn.py b/code/machine_learning_models/knn.py
index 4c72314308fbd294c464e6da12c1574d424c3311..e79b5236b84513636542caab2615157a59ce1790 100644
--- a/code/machine_learning_models/knn.py
+++ b/code/machine_learning_models/knn.py
@@ -2,14 +2,13 @@ import warnings
 import numpy as np
 import pandas as pd
 import seaborn as sns
-from sklearn.metrics import classification_report, confusion_matrix, roc_curve, auc
+from sklearn.metrics import classification_report, confusion_matrix
 from sklearn.neighbors import KNeighborsClassifier
 from sklearn.preprocessing import StandardScaler, LabelEncoder
 from sklearn.pipeline import Pipeline
 from sklearn.model_selection import train_test_split
 
-from utilities import ordinal_encode, normalize, plot_confusion_matrix, plot_counts, import_data, save_plot, show_plots, \
-    plot_roc_curve
+from utilities import ordinal_encode, normalize, plot_confusion_matrix, plot_counts, import_data, plot_roc_curve
 
 warnings.filterwarnings("ignore")
 
diff --git a/code/machine_learning_models/random_forest.py b/code/machine_learning_models/random_forest.py
index f050a6b5f6f1c0ed30fabf00ecbf6a8d5efcfae0..2e5056d19387a4ea1f82afff1b490786ff516fcd 100644
--- a/code/machine_learning_models/random_forest.py
+++ b/code/machine_learning_models/random_forest.py
@@ -1,10 +1,9 @@
 import numpy as np
 import pandas as pd
-from scipy.io import arff
 
 import warnings
 from sklearn.preprocessing import StandardScaler, LabelEncoder
-from sklearn.model_selection import train_test_split, learning_curve
+from sklearn.model_selection import learning_curve
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.metrics import classification_report, confusion_matrix, precision_recall_curve
 
diff --git a/code/machine_learning_models/resulting_figures/Decision Tree - Confusion Matrix.png b/code/machine_learning_models/resulting_figures/Decision Tree - Confusion Matrix.png
index 1c1d4be374ccb8199e0ddaff5a4d083ad611cd38..1873a44e159782b430503858bd395f7b1a0ddc01 100644
Binary files a/code/machine_learning_models/resulting_figures/Decision Tree - Confusion Matrix.png and b/code/machine_learning_models/resulting_figures/Decision Tree - Confusion Matrix.png differ
diff --git a/code/machine_learning_models/resulting_figures/Decision Tree - Feature Importance.png b/code/machine_learning_models/resulting_figures/Decision Tree - Feature Importance.png
index 87ded7e94e692a8a52d1b551ed9027d7d6059a3a..4ec659e1819a1084c7912f5885b9442f58bd3566 100644
Binary files a/code/machine_learning_models/resulting_figures/Decision Tree - Feature Importance.png and b/code/machine_learning_models/resulting_figures/Decision Tree - Feature Importance.png differ