Fixing pep8