Update data science solution

Overall, this test depends very heavily on the particular data within `source_data.json`, and it shouldn't. You should find a way to represent the data within the test file, so that it's easier to read.