Minor changes to example