minor updates to demo script
merging to repo