Train test split (17/33)

Train test split