diff --git a/modules/ml-pipeline/src/pipeline/configs/build_model.yaml b/modules/ml-pipeline/src/pipeline/configs/build_model.yaml
index 4c72487..354b2ca 100644
--- a/modules/ml-pipeline/src/pipeline/configs/build_model.yaml
+++ b/modules/ml-pipeline/src/pipeline/configs/build_model.yaml
@@ -13,7 +13,7 @@ default:
       output_filepath: ./data/model/allmodels/
       problem_type: regression
       eval_metric: mean_squared_error #mean_absolute_error
-      time_limit: 400
+      time_limit: 800
       presets: medium_quality
       excluded_model_types: ['KNN', 'RF']
       infer_limit: 0.05
diff --git a/modules/ml-pipeline/src/pipeline/dvc.lock b/modules/ml-pipeline/src/pipeline/dvc.lock
index 19173d2..4d669b5 100644
--- a/modules/ml-pipeline/src/pipeline/dvc.lock
+++ b/modules/ml-pipeline/src/pipeline/dvc.lock
@@ -29,8 +29,8 @@ stages:
     outs:
     - path: data/prepared_data/
       hash: md5
-      md5: 1b1f7467e4abc12e6febbf2a84756914.dir
-      size: 39780684
+      md5: 12b7939c38b6a201063b063ed64d521b.dir
+      size: 39840424
       nfiles: 2
   build_model:
     cmd: python 2_build_model.py
@@ -41,8 +41,8 @@ stages:
       size: 4149
     - path: data/prepared_data
       hash: md5
-      md5: 1b1f7467e4abc12e6febbf2a84756914.dir
-      size: 39780684
+      md5: 12b7939c38b6a201063b063ed64d521b.dir
+      size: 39840424
       nfiles: 2
     params:
       configs/build_model.yaml:
@@ -58,7 +58,7 @@ stages:
               output_filepath: ./data/model/allmodels/
               problem_type: regression
               eval_metric: mean_squared_error
-              time_limit: 400
+              time_limit: 800
               presets: medium_quality
               excluded_model_types:
               - KNN
@@ -68,12 +68,12 @@ stages:
     outs:
     - path: data/model/
       hash: md5
-      md5: c83b4cf0c51bd433bfb38307e978ed39.dir
-      size: 344485548
-      nfiles: 24
+      md5: 7d062363a9de5a659df638de1541d9ee.dir
+      size: 383515358
+      nfiles: 26
     - path: metrics/fit_metrics.json
       hash: md5
-      md5: 3105f9cf71b69b5b0f5675b2c169273c
+      md5: 06c50da7ca7fdb631896790b76a5e19d
       size: 223
   generate_predictions:
     cmd: python 3_generate_predictions.py
@@ -84,13 +84,13 @@ stages:
       size: 2464
     - path: data/model
       hash: md5
-      md5: c83b4cf0c51bd433bfb38307e978ed39.dir
-      size: 344485548
-      nfiles: 24
+      md5: 7d062363a9de5a659df638de1541d9ee.dir
+      size: 383515358
+      nfiles: 26
     - path: data/prepared_data
       hash: md5
-      md5: 1b1f7467e4abc12e6febbf2a84756914.dir
-      size: 39780684
+      md5: 12b7939c38b6a201063b063ed64d521b.dir
+      size: 39840424
       nfiles: 2
     params:
       configs/settings.yaml:
@@ -102,8 +102,8 @@ stages:
     outs:
     - path: data/predictions/
       hash: md5
-      md5: f914cf31400e228ee6e1386155b68e7c.dir
-      size: 369783
+      md5: d6c97ad17146677fe705ccd7bcbb4873.dir
+      size: 369475
       nfiles: 1
   generate_metrics:
     cmd: python 4_generate_metrics.py
@@ -114,13 +114,13 @@ stages:
       size: 3484
     - path: data/predictions
       hash: md5
-      md5: f914cf31400e228ee6e1386155b68e7c.dir
-      size: 369783
+      md5: d6c97ad17146677fe705ccd7bcbb4873.dir
+      size: 369475
       nfiles: 1
     - path: data/prepared_data
       hash: md5
-      md5: 1b1f7467e4abc12e6febbf2a84756914.dir
-      size: 39780684
+      md5: 12b7939c38b6a201063b063ed64d521b.dir
+      size: 39840424
       nfiles: 2
     params:
       configs/settings.yaml:
@@ -130,8 +130,8 @@ stages:
     outs:
     - path: metrics/metrics.json
       hash: md5
-      md5: c23b7f0628473bf42eef126167e8928e
-      size: 224
+      md5: 6bb037ff29c7119576c8818b395d32f6
+      size: 225
   startup_cleanup:
     cmd: python 0_startup_cleanup.py
     deps: