Handle CLI errors and report error message back to user

1 files changed, 6 insertions, 2 deletions

diff --git a/CLI/ruby/bin/game-of-life b/CLI/ruby/bin/game-of-life
index b490e1a..f83d72b 100755
--- a/CLI/ruby/bin/game-of-life
+++ b/CLI/ruby/bin/game-of-life
@@ -60,6 +60,10 @@ begin
   CLI.start(ARGV)
 rescue SystemExit, Interrupt
   "Do nothing when user exits with Interrupt (CMD/Ctrl + C)"
-rescue NotImplementedError
-  puts "Seed option isn't implemented yet, please use --input option"
+rescue NotImplementedError => e
+  puts e.message
+  exit(-1)
+rescue GameOfLife::Error => e
+  puts e.message
+  exit(-1)
 end