Fixing pep8 formatting