working on exercise 4