Update main.js

Added mockup button functionality.