Minor changes for demo.