minor change to demo and to docs