Improve code style (PEP8 compliance)