Update polish.py with comments