Small tweaks to example