Update data science solution

This is another awkward usage. If you have to grab elements by array index and then manipulate them manually, there's a good chance you're not returning the data you want. Generally, provide the final form of the data on the way out of the object.