Updated docs
Docstring fixes