Complete section 4 of lesson 7: Refactoring