Update search.py

Adding search