Update search.py

Committed changes to search