Code for assignment 2, Exploratory Data Analysis course