Submitting code for Assignment 2