Small cleanup to example