Updated demo

Updated demo