fixing exercise 4