added train/test data