Created for the data science class