Add docstrings and unbreak code