Finish HW2 Assignment