Train test split (21/33)

Train test split