Tweak class 6 exercise 4