Update to support new discreteRV interface