For my data science class codes