Updated the shell script