Added wait_for_token implementation