Add death and level restarting