Assignment 2 code to Exploratory Data Analysis course