pep8 work
pep8 work on cleaning up the code