Updating docs
Polish docstring