Add docstrings
Flake8