add back basic wrap logic to the shell script

Should not have removed this - over zealous on my part.

Signed-off-by: Jeremy C. Andrus <jeremya@cs.columbia.edu>