Tidy up main.js