Update main.js

match by class vs id to make it smaller