main.js clean up