Tidy up code.