minor changes to main.js file