Tidy up codes