Update keep_running.py