Update the demo page

- fixed some minor bug