Update demo
Fix demo