Add 'if's to run.sh