improve run.py
ok