Add initialization  and restart game logic