Updates to demo and documentation