Cleanup of example
Minor tweaks.