adjusted code to PEP-8 code compliance