Rescale both train and test data in submit() function.