Tidy up code slightly