Update worker-init.sh

Updated script for ubuntu images