add docstrings for classes
edits