update docstring, add function import