Correcting issues, HW7 extra credit