Tweak class 5 exercise 3

:fire_engine: