Add search_class function