main.js fixes