Small changes to demo.