Added some extra functionality to the login screen