Conformed problem 2 to PEP-8 standard