Update docs
Update docstrings