small changes to demo