Finish up adding docstrings to all public classes/methods