Assignment for data science course