minor changes to demo file