Update example to work with latest version of the library