edited code for PEP8 test