Code to complete lesson 16