Refactor 3 exercise 7 week 4