Some small pep8 corrections.
python 3 compatibility