Finished data science exercise