Programming assignment from Data Science