Tweak class 5 exercise 3