Modify simple.py