some changes for exercise 3