Small fixes to demo files