Update switch_after_exit.py