Update main.js

Highlightted the pressed button.