updated demo code.