chore(demo): clean up demo code