updated the demo script