Update atexit.py