diff --git a/TrainerClass.py b/TrainerClass.py
index de8a750..ff14848 100644
--- a/TrainerClass.py
+++ b/TrainerClass.py
@@ -236,10 +236,10 @@ class eNoseTrainer:
             'dense_units': tune.choice([32, 64, 128, 256]),
             'dropout': tune.choice([0.05, 0.15, 0.3]),
             'lr': tune.choice([0.01, 0.005, 0.001]),
-            'batch_size': tune.choice([16, 32, 64]),
+            'batch_size': tune.choice([16, 32, 64, 128]),
             'epochs': epochs
         }
-        total_space = (3*3*3*2*3*3*3*3)
+        total_space = (3*3*3*2*4*3*3*4)
 
         self.logger.info(f"total_space: {total_space}, num_samples: {int(nsamples*total_space)}")