HW4 correct
HW4 correct 2