Adding token_key option