Minor updates to demo.