Modify code for PEP8