Making corrections to exercise 4