Coursework for Data Science class