fix docstring, list comprehension for all_samples