pep 8 editing

Some small pep8 edits