it's for data science course home work