diff --git a/code/machine_learning_models/decision_tree.py b/code/machine_learning_models/decision_tree.py
index 98df192d4fc27d409f2f3c7cde749541252d77ce..32ff0fbb30f04ebc3c60fb506e7adaccfa570de5 100644
--- a/code/machine_learning_models/decision_tree.py
+++ b/code/machine_learning_models/decision_tree.py
@@ -96,7 +96,7 @@ train(dtc2)
 features = pd.DataFrame(dtc2.feature_importances_,
                         index= X.columns,
                         columns=['Importance']).sort_values(by='Importance', ascending=False)
-plot_features(features, model_name=model_name)
+plot_features(features, model_name=model_name + "2")
 
 def predict(prediction_input):
     if len(prediction_input) == 0:
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 4998c97a27b808354776000a021d55a56e50ede3..6937367aed106062468ead78a3ff5e32cf7fdf79 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
diff --git a/code/machine_learning_models/resulting_figures/Decision Tree2 - Feature Importance.png b/code/machine_learning_models/resulting_figures/Decision Tree2 - Feature Importance.png
new file mode 100644
index 0000000000000000000000000000000000000000..4998c97a27b808354776000a021d55a56e50ede3
Binary files /dev/null and b/code/machine_learning_models/resulting_figures/Decision Tree2 - Feature Importance.png differ