Tidy up js