add button + function to restart game after round