Added code for lesson 7