Bringing the code to the standard PEP 8