Add user_token related code