Modify login controller