Docstrings for the classes would be helpful (anything at this point would be better than none).