still working on the highlight feature