Update main.js

selector change