JS tidyup