update vignette and code per suggestions