add train/test data