Train test split (6/33)

Train test split