Update undo.py