Assignment for specialisation project in Coursera Data Science course