save train and test data