prepare_data_for_modelling function

Prepare Data for Training a model