Added method prepare_data_post_feature_selection