Worked more on homework 5