created for data science course assignment