Added game restart functionality