fixing HW2