Task 6 - small changes