Improve main.js