Clean up main.js