minor changes to demo