Add support for API keys