Add a few more docstrings for completeness