for data science homework