Changes to main.js