Small changes to demo code.