updated train/test split