Update restart_db.py