fixing highlight class