refactor
Working on challenge 10.