Updated docs to reflect recent API changes.
Started cleaning up docstrings.