Update js for highlight feature