for my data science class