Adapt to PEP8
Merge pull request #5 from changyuheng/pr_adapt_to_pep8

Adapt to PEP8