Code from coursera data science course