Updated the demo

Updated the demo