Solution to lesson 4