add docstring, remove unused import