Formatted  code (closer) to PEP8