Tidy up code changes