add data prep for training data