Small adjustments to exercise 4