clean up JS abit