Clean up demo code.