Added prompt for API key in unit test