diff --git a/lib/team_henk_marielle_hester_twee.rb b/lib/team_henk_marielle_hester_twee.rb index d231108..01af6df 100644 --- a/lib/team_henk_marielle_hester_twee.rb +++ b/lib/team_henk_marielle_hester_twee.rb @@ -44,12 +44,13 @@ class TeamHenkMarielleHesterTwee def initialize(roman, mode = 'arabic') @roman = roman + @mode = mode end [:+, :-, :*, :**, :/, :%].each do |op| define_method op do |other| res = self.to_i.send(op, other.to_i) - mode == 'arabic' ? res : self.class.to_roman(res) + @mode == 'arabic' ? res : self.class.to_roman(res) end end