Add API key support