Update replay.py