Adding docstrings and extra PEP-8 conformity