Pep8 fixes for python code