Minor improvement to example