Tweak main js file