Update restart.sh to make it comfier.