First answer for HW 2