Add vignette improvements from @tbates.

Closes #308