Minor improvements to example