Update example for the latest implementation