Added reset after end of game