Added 2 docstrings. Cleaned imports and pep8 syntax