improve how the code handles xarray ufunc broadcasting of pvalues