Update cleanup.py