Review docstrings and organization

Fixed a typo; added missing docstrings; a few stylistic changes.