diff --git a/Working Models/predict_from_json.py b/Working Models/Trained_Regression.py
similarity index 98%
rename from Working Models/predict_from_json.py
rename to Working Models/Trained_Regression.py
index d02afffc9422c137d2fcab9c83127f212c72eed8..a3818df915c9f57d39da7cf67c6d3bff2dc68838 100644
--- a/Working Models/predict_from_json.py	
+++ b/Working Models/Trained_Regression.py	
@@ -23,7 +23,7 @@ imputer = load(imputer_path)
 # Load Metadata
 with open(meta_path, "r") as f:
     metadata = json.load(f)
-target_col = metadata["target_column"]
+target_col = f"{sys.argv[1]}"
 
 # Load and preprocess data
 df = pd.read_csv(csv_path)