Update search_sample.py