Fix bugs, add demo