diff --git a/experiments/fmnist.jl b/experiments/fmnist.jl
index 7802e19d6a4cda064d45340692708e807b8bf0bc..34f602bfac44a77297bfb4e7e0f3c978383ca8d4 100644
--- a/experiments/fmnist.jl
+++ b/experiments/fmnist.jl
@@ -62,6 +62,9 @@ params = (
     dim_reduction = true,
 )
 
+# Best grid search params:
+params = append_best_params(params, dataname)
+
 if GRID_SEARCH
     grid_search(
         counterfactual_data,
diff --git a/experiments/jobscripts/generators/fmnist.sh b/experiments/jobscripts/generators/fmnist.sh
index b4ea63f1b119947e7f96f53b398a1977d4a8f66c..8969365350864126bd915929c4aadf2c548df530 100644
--- a/experiments/jobscripts/generators/fmnist.sh
+++ b/experiments/jobscripts/generators/fmnist.sh
@@ -2,7 +2,7 @@
 
 #SBATCH --job-name="Fashion MNIST (ECCCo)"
 #SBATCH --time=05:00:00
-#SBATCH --ntasks=40
+#SBATCH --ntasks=50
 #SBATCH --cpus-per-task=10
 #SBATCH --partition=compute
 #SBATCH --mem-per-cpu=10GB
@@ -13,4 +13,4 @@ module load 2023r1 openmpi
 
 source experiments/slurm_header.sh
 
-srun julia --project=experiments --threads $SLURM_CPUS_PER_TASK experiments/run_experiments.jl -- data=fmnist output_path=results mpi threaded n_individuals=50 n_runs=10 > experiments/logs/fmnist.log
+srun julia --project=experiments --threads $SLURM_CPUS_PER_TASK experiments/run_experiments.jl -- data=fmnist output_path=results mpi threaded n_individuals=100 n_runs=5 > experiments/logs/fmnist.log
diff --git a/experiments/jobscripts/generators/mnist.sh b/experiments/jobscripts/generators/mnist.sh
index 4c79e8b10ada123e04f46c068cb346b20209f672..58b35400f889624110782873a364889673e01a13 100644
--- a/experiments/jobscripts/generators/mnist.sh
+++ b/experiments/jobscripts/generators/mnist.sh
@@ -2,10 +2,10 @@
 
 #SBATCH --job-name="MNIST (ECCCo)"
 #SBATCH --time=05:00:00
-#SBATCH --ntasks=40
+#SBATCH --ntasks=50
 #SBATCH --cpus-per-task=10
 #SBATCH --partition=compute
-#SBATCH --mem-per-cpu=10GB
+#SBATCH --mem-per-cpu=12GB
 #SBATCH --account=research-eemcs-insy
 #SBATCH --mail-type=END     # Set mail type to 'END' to receive a mail when the job finishes. 
 
@@ -13,4 +13,4 @@ module load 2023r1 openmpi
 
 source experiments/slurm_header.sh
 
-srun julia --project=experiments --threads $SLURM_CPUS_PER_TASK experiments/run_experiments.jl -- data=mnist output_path=results mpi threaded n_individuals=50 n_runs=10 > experiments/logs/mnist.log
+srun julia --project=experiments --threads $SLURM_CPUS_PER_TASK experiments/run_experiments.jl -- data=mnist output_path=results mpi threaded n_individuals=100 n_runs=5 > experiments/logs/mnist.log
diff --git a/experiments/mnist.jl b/experiments/mnist.jl
index 782811cc5be4e5603fb0deb585bb55fc7d3d2bc8..a4e9d50dc23500e891105fe001169dcdddd1ca99 100644
--- a/experiments/mnist.jl
+++ b/experiments/mnist.jl
@@ -62,6 +62,9 @@ params = (
     dim_reduction = true,
 )
 
+# Best grid search params:
+params = append_best_params(params, dataname)
+
 if GRID_SEARCH
     grid_search(
         counterfactual_data,