HW1Q5 code