Fix PEP8 issues with code formatting