Added login button to hadle login states better