remove unused import. Improve docstrings