improve code style to be compliant with PEP8