Uploaded the assignment 2 code