Main.js tweak