Add support for the staging environment