Update example. Fixes #4