Solution to lesson 2