Updated .sh file with modified code