Modify login function