Fixed syntax with pep8