Fix docstring
Add docstrings and clear up an assignment