add docstring and label argument