modifying clean_data again s.t. it can clean other things than just self.df_train