Adding data science solution

Good test cases - I like how you set up your own data set. Putting a blank line between each `it` clause will make the tests a little easier to read.