Clean JS up