Fix/Add docstrings