Add API token support