Improved docs
Improved docstrings