fix bugs & add a Demo