Update TC2_test.py