rearrange main.js code