coding style refine.

fix the coding style according to PEP8