diff --git a/main.py b/main.py
index aadf659..fb18190 100644
--- a/main.py
+++ b/main.py
@@ -6,12 +6,12 @@ from datetime import datetime, timezone, timedelta
 import calendar
 
 rf_reg_grid = {
-    'bootstrap': [True, False],
-    'max_depth': [5, 10, 70, 90, 100, None],
-    'max_features': ['auto', 'sqrt'],
+    'bootstrap': [True],
+    'max_depth': [5, 10, 20, 30, 40, 50, 60, 70, 80, 90, 100, None],
+    'max_features': ['auto', 'sqrt', 'log2'],
     'min_samples_leaf': [1, 2, 4, 6],
-    'min_samples_split': [2, 5, 10, 12],
-    'n_estimators': [100, 200]
+    'min_samples_split': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12],
+    'n_estimators': [80, 90, 100, 120, 150, 200]
 }
 
 best_params = {
@@ -132,7 +132,7 @@ def main():
     # Prepare data for moddeling
     df = process_data(df=df)
     # Split data into train and test sets
-    X_train, X_test, y_train, y_test = split_data(df=df, split_value=0.82) 
+    X_train, X_test, y_train, y_test = split_data(df=df, split_value=0.95) 
 
     # Model the data
     model = find_best_hyperparameters_and_train(