Update simple_test.py