Small fixes for run.sh
Merge pull request #4 from afronski/master

Small fixes