Small changes in main.js