Restructure main.js