Updated login code