Fix docs
Improve docs
Improve docs
Fix doctest