Updated file containing Assignment 2 solution