Add dataset search functionality