Added a few pep8 fixes