Added code for lesson 2