added changes to main.js