exercise 1 - problem 2 fix