Add some docstrings, not finished