Minor edits to main.js