Update test_example.py