Update search_space.py