Create prediction_main.R