Update simple_example.py