Fixing docstrings and imports not used.