Assignment for data science - Coursera program