Tweaks to example