tidy up JS