flake8
Add docstrings