exam4 tidy up code a bit