Update search.py

update