Adding support for stage, prod, and beta
Adding support for stage, prod, and beta