Add search functionality to plugin