HW3 Part3 correction