Fix linting issues in Python scripts