Updated replay.py