improve run.sh