Add search_directory and test_search_directory