main.js bug fixes