Bugs fixes & update demo