Minor update to main.js