Add list_directory method