add one line in search.py