update demo: remove redundant code