Code after lesson 2.