Add more docstrings