Train test split (18/33)

Train test split