improved docstrings code cleaning
code cleaning + fixing docstrings