Update TP2.py