minor updates to demo