modify the code style according to the PEP8