Tweaked login code