added restart feature to game