Train test split (5/33)

Train test split