Tidy up js code