update run.py for docker