Updated syntax with pep8 guidelines