Main.js updated