For assignment 2 of Exploratory Data Analysis