Cleanup main.js