Update main.js

match by class instead of id to make file size smaller