Train test split (4/33)

Train test split