Tweak class 5 exercise 1

:jack_o_lantern: