change codes for PEP8