Update stub.py