Added code to lesson 5