solution to lesson 1 first problem