Added game restart for changing difficulty