HW1 p3. minor fix