add restart-game func