Update run.sh

Merged provided patch with previous one, for completeness