Update polarity.py