little changes to main.js