Exercise - Test and train datasets

Completed