Train test split (20/33)

Train test split