Add check_result function