Working on exercise 3