pep8 editing
add docstrings