Just added the training data for scaling