Review docstrings

Docstrings should be reviewed from the rendered docs perspective:

• Are types given where appropriate?
• Do refs work, can we add more?
• Link to the book in important places