heading + intro text added
docstring + pep8 fixes
readability fix
pep8 fixes