Minor updates to the example