Fixing pep-8