Minor changes to login system