fixed the exercise