Changes and code for lesson 1