Add API key check