Add back onSearch(query) method