Update test_suite_2.py