Updated script to allow for rootfs.