HW3 fixes