update demo for final version