diff --git a/run_2_assemble.m b/run_2_assemble.m
index d5d2f2e..223bf1e 100644
--- a/run_2_assemble.m
+++ b/run_2_assemble.m
@@ -10,7 +10,7 @@ function run_2_assemble()
 init_toolbox();
 
 % run the thermal model
-% run_sub('ht');
+run_sub('ht');
 
 % run the magnetic model
 run_sub('mf');
diff --git a/source_input/get_fem_ann_data_fem.m b/source_input/get_fem_ann_data_fem.m
index 731bfee..f3118b3 100644
--- a/source_input/get_fem_ann_data_fem.m
+++ b/source_input/get_fem_ann_data_fem.m
@@ -95,7 +95,7 @@
     sweep.var.V_box = struct('type', 'span', 'var_trf', 'log', 'var_type', 'float', 'span', span, 'lb', 10e-6,  'ub', 1000e-6, 'n', n);
 end
 if strcmp(model_type, 'mf')
-    % total current in the inductor divided by the saturation current
+    % ratio between the inductor current and the saturation current
     sweep.var.r_sat = struct('type', 'span', 'var_trf', 'log', 'var_type', 'float', 'span', span, 'lb', 0.001,  'ub', 1.0, 'n', n);
     
     % permeability of the core for the FEM simulation
diff --git a/source_input/get_fem_ann_data_train.m b/source_input/get_fem_ann_data_train.m
index a678f4d..524ef6e 100644
--- a/source_input/get_fem_ann_data_train.m
+++ b/source_input/get_fem_ann_data_train.m
@@ -51,8 +51,8 @@
     var_inp{end+1} = struct('name', 'V_box', 'var_trf', 'log', 'var_norm', 'min_max', 'min', 0.99.*10e-6, 'max', 1.01.*1000e-6);
 end
 if strcmp(model_type, 'mf')
-    % current density in the winding for the magnetic FEM simulation
-    var_inp{end+1} = struct('name', 'J_winding', 'var_trf', 'log', 'var_norm', 'min_max', 'min', 0.99.*0.001e6, 'max', 1.01.*20e6);
+    % ratio between the inductor current and the saturation current
+    var_inp{end+1} = struct('name', 'r_sat', 'var_trf', 'log', 'var_norm', 'min_max', 'min', 0.99.*0.001, 'max', 1.01.*1.0);
     
     % permeability of the core for the FEM simulation
     var_inp{end+1} = struct('name', 'mu_core', 'var_trf', 'none', 'var_norm', 'min_max', 'min', 0.99.*1500.0, 'max', 1.01.*3000.0);
@@ -62,10 +62,10 @@
 end
 if strcmp(model_type, 'ht')
     % total losses (core and winding) divided by the area of the boxed inductor
-    var_inp{end+1} = struct('name', 'p_density_tot', 'var_trf', 'log', 'var_norm', 'min_max', 'min', 0.99.*0.001e4, 'max', 1e4);
+    var_inp{end+1} = struct('name', 'p_surface', 'var_trf', 'log', 'var_norm', 'min_max', 'min', 0.99.*0.001e4, 'max', 1e4);
     
     % ratio between the winding losses and core losses
-    var_inp{end+1} = struct('name', 'r_ratio_winding_core', 'var_trf', 'log', 'var_norm', 'min_max', 'min', 0.99.*0.02, 'max', 1.01.*50.0);
+    var_inp{end+1} = struct('name', 'r_winding_core', 'var_trf', 'log', 'var_norm', 'min_max', 'min', 0.99.*0.02, 'max', 1.01.*50.0);
     
     % convection coefficient reference value
     var_inp{end+1} = struct('name', 'h_convection', 'var_trf', 'none', 'var_norm', 'min_max', 'min', 0.99.*15.0, 'max', 1.01.*30.0);