minor changes for demo