Add more documentation in docstrings