improve cleanup() in run.sh