DOC: added docstring for function