Small fixes in the demo file