Code to complete lesson 17