improve search feature