Update main.js

New addition starting line 37-84 to accommodate for new buttons!