Add some pep8