Tidy up auth code