Add lesson 2 code