data science course homework