code in the data science course