assignment  for data science course