Update main.js

new code