diff --git a/experiments/spambase/stage1/params.json b/experiments/spambase/stage1/params.json
new file mode 100644
index 0000000000000000000000000000000000000000..df611626b3aa60533fe51fb3a68efb5e0497a1e7
--- /dev/null
+++ b/experiments/spambase/stage1/params.json
@@ -0,0 +1,28 @@
+{
+    "scorer": "accuracy",
+    "best_score_train": 0.9469882246376812,
+    "best_score_test": 0.9512124502352516,
+    "best_parameters": {
+        "min_samples_leaf": 1,
+        "n_estimators": 1000,
+        "max_features": "log2",
+        "max_depth": 20
+    },
+    "random_seed": [
+        1260,
+        1414,
+        1635,
+        513,
+        678,
+        943,
+        1496,
+        1434,
+        260,
+        1825,
+        415,
+        740,
+        811,
+        93,
+        1860
+    ]
+}
\ No newline at end of file
diff --git a/results/breast_cancer/stage4_fix/losses.png b/results/breast_cancer/stage4_fix/losses.png
index 577016a2ea1e40f9a026a7a7b47b839c98ae16da..65bd8f28f2b18dc24ade549b9c0927909d9824f7 100644
Binary files a/results/breast_cancer/stage4_fix/losses.png and b/results/breast_cancer/stage4_fix/losses.png differ
diff --git a/results/kr-vs-kp/stage4_fix/losses.png b/results/kr-vs-kp/stage4_fix/losses.png
index af17ea8990d8de60ad2d9b544a533a42b68c0770..08d7af6f6354d596f53a73afd924a72225506ee7 100644
Binary files a/results/kr-vs-kp/stage4_fix/losses.png and b/results/kr-vs-kp/stage4_fix/losses.png differ
diff --git a/results/steel-plates/stage4_fix/losses.png b/results/steel-plates/stage4_fix/losses.png
index 3a72cd7d2c0a674a272eb76b61295194faee64d5..f473402fc01d0f6d2d8c86941bae6862beb4fb04 100644
Binary files a/results/steel-plates/stage4_fix/losses.png and b/results/steel-plates/stage4_fix/losses.png differ
diff --git a/scripts/run_compute_results.sh b/scripts/run_compute_results.sh
index 74db5ec579d6df22cf7d5a8023d008253e3f629e..1e8cf514ad361f31cbcf204696a766503eed1782 100644
--- a/scripts/run_compute_results.sh
+++ b/scripts/run_compute_results.sh
@@ -1,4 +1,4 @@
-for dataset in steel-plates
+for dataset in breast_cancer kr-vs-kp steel-plates
 do
     python code/compute_results.py --stage=4 --experiment_ids 1 2 3 --dataset_name=$dataset --models_dir=models/$dataset/stage5
 done
diff --git a/scripts/run_stage5_experiments.sh b/scripts/run_stage5_experiments.sh
index 837231b6eb191b1651f955fe9408e77222655aca..93abbe76fe894b3ded7204114934e301958c0bd4 100644
--- a/scripts/run_stage5_experiments.sh
+++ b/scripts/run_stage5_experiments.sh
@@ -4,7 +4,7 @@ walltime=1:00
 seeds='1 2 3 4 5 6 7'
 
 #for dataset in breast_cancer diabetes diamonds california_housing boston linnerud steel-plates
-for dataset in steel-plates
+for dataset in breast_cancer kr-vs-kp steel-plates
 do
     oarsub -p "(gpu is null)" -l /core=$core_number,walltime=1:00 "conda activate test_env && python code/train.py --dataset_name=$dataset --seeds $seeds --extraction_strategy=none --extracted_forest_size_stop=1 --extracted_forest_size_samples=30 --experiment_id=1 --models_dir=models/$dataset/stage5 --subsets_used train+dev,train+dev"
     oarsub -p "(gpu is null)" -l /core=$core_number,walltime=1:00 "conda activate test_env && python code/train.py --dataset_name=$dataset --seeds $seeds --extraction_strategy=random --extracted_forest_size_stop=1 --extracted_forest_size_samples=30 --experiment_id=2 --models_dir=models/$dataset/stage5 --subsets_used train+dev,train+dev"