Update main.js

Fix #13