Update example code to reflect current usage

See #23