Python Code Quality Improvement