Update expect.py