Update example to reflect #3

Closes #4