Edits to demo.js