Refactor 1 for challenge 2 week 4