Train test split (16/33)

Train test split