Worked on bonus problem