Code for data science project from coursera