Worked on challenge 2