Changes to example to make it work with package