minor fixes to demo