Small changes to bash script