Computing for Data Analysis, assignment 2