Fixing syntax to comply with pep8