adding to exercise 3