Changes to exercise 4
Updated exercise 4