progress on train/test split