minor edits for demo