Updates to demo