Add check-login function