pep8 cleanup
pep8 + adding needed functions