Update problem4_polin.py