Add restart game functionality