minor changes to main js