Working on search function
search and highlighting works (finally!)