Update test_exercise1.py