Cleaning up the example