Use toggleClass to add/remove the highlight class