Updating the example with recent changes