HW1-2: found bug