Update to init script for the worker nodes