Add docstrings and PEP8
Doc + PEP8