Refactor 1 exercise 3