Minor changes to the example