Part of project for Data Science class in Coursera