Simplify _add_back_token and _add_back_tokens