Add several new module docs and clean up some docstrings