Update docs and fix some docstring mistakes