Change the code to comply with PEP 8