Modify excercise 1