diff --git a/32 Diamond/ruby/diamond.rb b/32 Diamond/ruby/diamond.rb index 1c8493a3..251cee25 100644 --- a/32 Diamond/ruby/diamond.rb +++ b/32 Diamond/ruby/diamond.rb @@ -11,9 +11,7 @@ end def get_facets while true number = gets.chomp - if /^\d+$/.match(number) - return number.to_i - end + return number.to_i if /^\d+$/.match(number) puts "!NUMBER EXPECTED - RETRY INPUT LINE" print "? " end @@ -42,4 +40,6 @@ def main draw_diamonds(lines) end +trap "SIGINT" do puts; exit 130 end + main \ No newline at end of file