For a data science class