Added more code in main.js