More changes to problem 5 section 6.1 homework