Updated end search callback name and added on_begin_search callback. Also updated tests to pass.