Update the init script used in docker