HW2 correct