Fixed error in HW2