Code added to login