Code formatting cleanup - PEP 8