Add search highlighting.
Improve search functionality.