For the data science class