Add more functionality to login