1 files changed, 6 insertions, 8 deletions
diff --git a/go.py b/go.py
index 548df38..cf0a673 100755
--- a/go.py
+++ b/go.py
@@ -2,29 +2,27 @@
 
 """Play Go!"""
 
-from imago.engine.parseHelpers import ParseCodes, parseVertex
+from imago.engine.parseHelpers import parseVertex
 from imago.gameLogic.gameState import GameState
 
 if __name__ == "__main__":
 
-    #GAMESTATE = GameState(5)
-    GAMESTATE = GameState(19)
+    GAMESTATE = GameState(9)
 
-    while 1:
+    while True:
 
         GAMESTATE.getBoard().printBoard()
 
         move = input("Move (" + GAMESTATE.getPlayerCode() + "): ")
-        move = parseVertex(move, GAMESTATE.size)
 
-        if move == ParseCodes.ERROR:
+        try:
+            move = parseVertex(move, GAMESTATE.size)
+        except Exception as err:
             print("Invalid move syntax. Example of move: A1")
             continue
 
         print()
 
-        player = str(GAMESTATE.getPlayerCode())
-
         moveRow = move[0]
         moveCol = move[1]