Update restart.py