Bug fixes, improved testing and vignette added