Tidy up code order