Update example to reflect current API