Update polish.py