Pep 8 clean up

Just some pep8 clean up for now.