adjusted code for PEP 8