small changes for exercise 3.5