Updated the demo script