Create Final Version of HW2 Code