Update search.py

Added comments