Add a highlighted class once highlighing is done