chain.rb

class Chain < Struct.new(:value)
  def method_missing(method, *args, &block)
    next_value = self.value.public_send(method, *args, &block)
    self.value = next_value unless method.to_s[-1] == "!"
    self
  end
end

# Now you can chain bang methods (or regular methods)!
Chain.new([1, 2, 3, 4, 5]).reject!(&:odd?).reverse!.value # => [4, 2]

Return self after the if/else

Fixed 👍