diff --git a/doc/hyperparametersOptimisation/min_samples_leaf_max_features.png b/doc/hyperparametersOptimisation/min_samples_leaf_max_features.png
index 83119a5..c793661 100644
Binary files a/doc/hyperparametersOptimisation/min_samples_leaf_max_features.png and b/doc/hyperparametersOptimisation/min_samples_leaf_max_features.png differ
diff --git a/src/training/decision_tree/main.py b/src/training/decision_tree/main.py
index 0d30666..b4ece58 100644
--- a/src/training/decision_tree/main.py
+++ b/src/training/decision_tree/main.py
@@ -106,7 +106,7 @@ if __name__ == "__main__":
             "label": "Max features",
             "values": [
                 round(value, 2)
-                for value in np.arange(0.2, 0.5, 0.05)
+                for value in np.arange(0.45, 1, 0.05)
             ]
         }
         base_model_parameters = {
@@ -129,7 +129,7 @@ if __name__ == "__main__":
 
         # Plot graph
         fig, ax = plt.subplots()
-        scatter = ax.scatter([], [], c=[], cmap="RdYlGn_r", vmin=0, vmax=2)
+        scatter = ax.scatter([], [], c=[], cmap="RdYlGn_r", vmin=1.3, vmax=1.6)
         colorbar = fig.colorbar(scatter, ax=ax)
         colorbar.set_label("Test MAE")
         ax.set_xlabel(parameter_1["label"])