Fix to class 7 exercise 1