Added game restart after death