fixing exercise 2

some changes