Clean up the demo code.