Add cleanup to the example

closes #2