Add docstrings and format code