Update main.js

Update for #1