Add basic support for staging environment