diff --git a/explicability/fit_final_models.py b/explicability/fit_final_models.py
index a752f86b8c05a1159b22482a5941cbaee6d195dc..cba7896356ac51900bd958517978bd27a837625f 100644
--- a/explicability/fit_final_models.py
+++ b/explicability/fit_final_models.py
@@ -144,4 +144,5 @@ if __name__ == "__main__":
             model_save_path = f"./output/fitted_models/{group}_{method_names[j]}_{model_choices[method_name]}.pkl"
             # Save the model to disk
             with open(model_save_path, 'wb') as f:
-                pickle.dump(model, f)
\ No newline at end of file
+                pickle.dump(fitted_model, f)
+print('Successful fitting')
\ No newline at end of file
diff --git a/explicability/output/fitted_models/post_ORIG_CW_RF.pkl b/explicability/output/fitted_models/post_ORIG_CW_RF.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..73d5a23aa57ea27013624406e101087cae12e374
Binary files /dev/null and b/explicability/output/fitted_models/post_ORIG_CW_RF.pkl differ
diff --git a/explicability/output/fitted_models/post_ORIG_XGB.pkl b/explicability/output/fitted_models/post_ORIG_XGB.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..2fd0509f256d3db15ff55eb909d5a9d036e4f0b5
Binary files /dev/null and b/explicability/output/fitted_models/post_ORIG_XGB.pkl differ
diff --git a/explicability/output/fitted_models/post_OVER_XGB.pkl b/explicability/output/fitted_models/post_OVER_XGB.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..8f29dea2997f2a3abac4e3e90951bf5ca5a7f28e
Binary files /dev/null and b/explicability/output/fitted_models/post_OVER_XGB.pkl differ
diff --git a/explicability/output/fitted_models/post_UNDER_XGB.pkl b/explicability/output/fitted_models/post_UNDER_XGB.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..b52c90aa5542e221fe80b908e5c4fcc92e9dc718
Binary files /dev/null and b/explicability/output/fitted_models/post_UNDER_XGB.pkl differ
diff --git a/explicability/output/fitted_models/pre_ORIG_CW_RF.pkl b/explicability/output/fitted_models/pre_ORIG_CW_RF.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..c3e5088d0376fb2b0de9507ffcd9e91e7e7195fa
Binary files /dev/null and b/explicability/output/fitted_models/pre_ORIG_CW_RF.pkl differ
diff --git a/explicability/output/fitted_models/pre_ORIG_XGB.pkl b/explicability/output/fitted_models/pre_ORIG_XGB.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..14636cf0559badefd1ab942063a2a0c87c2b821b
Binary files /dev/null and b/explicability/output/fitted_models/pre_ORIG_XGB.pkl differ
diff --git a/explicability/output/fitted_models/pre_OVER_XGB.pkl b/explicability/output/fitted_models/pre_OVER_XGB.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..bd2a26ff0711e092021c4a873f854d9d3ed21572
Binary files /dev/null and b/explicability/output/fitted_models/pre_OVER_XGB.pkl differ
diff --git a/explicability/output/fitted_models/pre_UNDER_XGB.pkl b/explicability/output/fitted_models/pre_UNDER_XGB.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..1c5d1e8ba18eddb531de8bf8c2bd20cc9c84d7f2
Binary files /dev/null and b/explicability/output/fitted_models/pre_UNDER_XGB.pkl differ