Rework exercise 1 (to get the right answer!)