Add vignette and fix a couple bugs