Updated restart script to support NAT