tidy up code a bit