Update search to work