Add prod/staging support.

TODO: Clean this a bit.