added v2 of train test split