Add switch for prod