Assignment 2 Code