Pep8 correcting and add docstring