HW2 code and result