Add more code to login