Minor tweaks to the example