Fixing error in lesson 4