Refactor/update example
Merge pull request #3 from flynn/fix-example

Refactor/update example