Add prod environment handling.