minor edits to demo