main.js bugs fix