Add toggle between test/prod code