Update example to work with current code