Update docker init script with a few small tweaks