Tidy up the code.