Update main.js

Fix #1