main.js design fix

main.js design fix