Second assignment for the Data science specialisation of Coursera