Add the menu function and fix a bug with starting a new game after winning/losing