Update code to correspond rwfs & correct vignette