Added the correct code for lesson 1