Add support for staging environment