Updated search.py