small changes to exercise 2c