Added some pep8