Tweak class 5 exercise 2

:dancer: