Updated login/token code