Update sub_search.py