update demo to clean up