From 056c8cd4d971174bd68aed81708a2de1b39e59a2 Mon Sep 17 00:00:00 2001
From: Thomas Kwashnak <ThomasKwashnak@gmail.com>
Date: Thu, 6 Jan 2022 09:06:15 -0500
Subject: [PATCH] Update HighIQ.java

---
 48_High_IQ/java/src/HighIQ.java | 36 ++++++++++++++++++++++++++++-----
 1 file changed, 31 insertions(+), 5 deletions(-)

diff --git a/48_High_IQ/java/src/HighIQ.java b/48_High_IQ/java/src/HighIQ.java
index 93340556..77e0e5ae 100644
--- a/48_High_IQ/java/src/HighIQ.java
+++ b/48_High_IQ/java/src/HighIQ.java
@@ -62,12 +62,38 @@ public class HighIQ {
     }
 
     public void play() {
-        while(true) {
-            while(!move()) {
-                System.out.println("ILLEGAL MOVE, TRY AGAIN...");
-
+        do {
+            do {
+                while(!move()) {
+                    System.out.println("ILLEGAL MOVE, TRY AGAIN...");
+                }
+            } while(!isGameFinished());
+            
+            int pegCount = 0;
+            for(Integer key : board.getKeySet()) {
+                if(board.getOrDefault(key,false)) {
+                    pegCount++;
+                }
             }
-        }
+        
+            out.println("YOU HAD " + pegCount + " PEGS REMAINING");
+        
+            if(pegCount == 1) {
+                out.println("BRAVO!  YOU MADE A PERFECT SCORE!");
+                out.println("SAVE THIS PAPER AS A RECORD OF YOUR ACCOMPLISHMENT!");
+            }
+            
+        } while(playAgain());
+    }
+
+    private boolean playAgain() {
+        out.println("PLAY AGAIN (YES OR NO)");
+        return scanner.nextLine().toLowerCase().equals("yes");
+    }
+    
+    
+    public boolean isGameFinished() {
+        return false;
     }
 
     public void printBoard() {