diff --git a/modules/ml-pipeline/src/pipeline/configs/settings.yaml b/modules/ml-pipeline/src/pipeline/configs/settings.yaml
index 1d031cc..bcd5ee5 100644
--- a/modules/ml-pipeline/src/pipeline/configs/settings.yaml
+++ b/modules/ml-pipeline/src/pipeline/configs/settings.yaml
@@ -35,7 +35,9 @@ default:
       drop_columns: [
         "heat_demand_change", "carbon_change", "rdsap_change", "sap_ending", "carbon_ending", "days_to_starting", "days_to_ending",
         'number_habitable_rooms_starting', 'number_habitable_rooms_ending', 'number_heated_rooms_starting', 'number_heated_rooms_ending',
-        'number_habitable_rooms', 'number_heated_rooms', 'lighting_cost_starting', 'lighting_cost_ending', 'heating_cost_starting', 'heating_cost_ending', 'hot_water_cost_starting', 'hot_water_cost_ending',]
+        'number_habitable_rooms', 'number_heated_rooms', 'lighting_cost_starting', 'lighting_cost_ending', 'heating_cost_starting',
+        'heating_cost_ending', 'hot_water_cost_starting', 'hot_water_cost_ending',
+        'floor_thermal_transmittance', 'floor_thermal_transmittance_ending']
       # retain_features: ["SAP_STARTING", "TOTAL_FLOOR_AREA_DIFF"]
       retain_features: null
       # retain_features: ['uprn', 'sap_starting', 'hot_water_energy_eff_ending',
diff --git a/modules/ml-pipeline/src/pipeline/dvc.lock b/modules/ml-pipeline/src/pipeline/dvc.lock
index 06ecb77..3a05ca2 100644
--- a/modules/ml-pipeline/src/pipeline/dvc.lock
+++ b/modules/ml-pipeline/src/pipeline/dvc.lock
@@ -40,6 +40,8 @@ stages:
         - heating_cost_ending
         - hot_water_cost_starting
         - hot_water_cost_ending
+        - floor_thermal_transmittance
+        - floor_thermal_transmittance_ending
         default.feature_processor.feature_processor_config.retain_features:
         default.feature_processor.feature_processor_config.subsample_amount:
         default.feature_processor.feature_processor_config.subsample_seed: 0
@@ -55,8 +57,8 @@ stages:
     outs:
     - path: data/prepared_data/
       hash: md5
-      md5: a2a418c9b72ef168dcc94f8bca824527.dir
-      size: 41112441
+      md5: d2d8dcda26ce0f480b6bbf19ddb00693.dir
+      size: 39648081
       nfiles: 2
   build_model:
     cmd: python 2_build_model.py
@@ -67,8 +69,8 @@ stages:
       size: 4820
     - path: data/prepared_data
       hash: md5
-      md5: a2a418c9b72ef168dcc94f8bca824527.dir
-      size: 41112441
+      md5: d2d8dcda26ce0f480b6bbf19ddb00693.dir
+      size: 39648081
       nfiles: 2
     params:
       configs/build_model.yaml:
@@ -100,18 +102,18 @@ stages:
     outs:
     - path: data/fit_predictions/
       hash: md5
-      md5: a7df97430fe3f5eff6f15cb2d2431e50.dir
-      size: 3125410
+      md5: 6fbb13029ed64370b8686a863c15bc3d.dir
+      size: 3126163
       nfiles: 1
     - path: data/model/
       hash: md5
-      md5: f2dfac63cb8f16cf81ec7f058b38dc26.dir
-      size: 564562493
-      nfiles: 36
+      md5: 3e0a89879716a30edd964bb7cd727be4.dir
+      size: 704092080
+      nfiles: 35
     - path: metrics/fit_metrics.json
       hash: md5
-      md5: b87b67933d6489d508c4c72f1b543e85
-      size: 222
+      md5: 85a502c513759da26bfacd38aab7e114
+      size: 223
   generate_predictions:
     cmd: python 3_generate_predictions.py
     deps:
@@ -121,13 +123,13 @@ stages:
       size: 2464
     - path: data/model
       hash: md5
-      md5: f2dfac63cb8f16cf81ec7f058b38dc26.dir
-      size: 564562493
-      nfiles: 36
+      md5: 3e0a89879716a30edd964bb7cd727be4.dir
+      size: 704092080
+      nfiles: 35
     - path: data/prepared_data
       hash: md5
-      md5: a2a418c9b72ef168dcc94f8bca824527.dir
-      size: 41112441
+      md5: d2d8dcda26ce0f480b6bbf19ddb00693.dir
+      size: 39648081
       nfiles: 2
     params:
       configs/settings.yaml:
@@ -139,8 +141,8 @@ stages:
     outs:
     - path: data/predictions/
       hash: md5
-      md5: 4def629396860398d224fa175e33d90f.dir
-      size: 426281
+      md5: ae73bbac9406a542cf47be7065d303fd.dir
+      size: 426229
       nfiles: 1
   generate_metrics:
     cmd: python 4_generate_metrics.py
@@ -151,13 +153,13 @@ stages:
       size: 3447
     - path: data/predictions
       hash: md5
-      md5: 4def629396860398d224fa175e33d90f.dir
-      size: 426281
+      md5: ae73bbac9406a542cf47be7065d303fd.dir
+      size: 426229
       nfiles: 1
     - path: data/prepared_data
       hash: md5
-      md5: a2a418c9b72ef168dcc94f8bca824527.dir
-      size: 41112441
+      md5: d2d8dcda26ce0f480b6bbf19ddb00693.dir
+      size: 39648081
       nfiles: 2
     params:
       configs/settings.yaml:
@@ -167,8 +169,8 @@ stages:
     outs:
     - path: metrics/metrics.json
       hash: md5
-      md5: 63b75a4ae4c760df62597e7145f87452
-      size: 223
+      md5: 3f863e94d352afd6f814d32b95b2eb87
+      size: 221
   generate_scenerio_metrics:
     cmd: python 5_generate_scenarios.py
     deps: