Minor updates to main.js