Small changes, adding to vignette